jerryzh168 commited on
Commit
5abdf32
·
verified ·
1 Parent(s): 2591c62

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -61,9 +61,9 @@ def check_model_exists(
61
  if (
62
  quantization_type in ["Int4WeightOnly", "GemliteUIntXWeightOnly"]
63
  ) and (group_size is not None):
64
- repo_name = f"{username}/{model_name.split('/')[-1]}-ao-{MAP_QUANT_TYPE_TO_NAME[quantization_type.lower()]}-gs{group_size}"
65
  else:
66
- repo_name = f"{username}/{model_name.split('/')[-1]}-ao-{MAP_QUANT_TYPE_TO_NAME[quantization_type.lower()]}"
67
  if repo_name in model_names:
68
  return f"Model '{repo_name}' already exists in your repository."
69
  else:
 
61
  if (
62
  quantization_type in ["Int4WeightOnly", "GemliteUIntXWeightOnly"]
63
  ) and (group_size is not None):
64
+ repo_name = f"{username}/{model_name.split('/')[-1]}-ao-{MAP_QUANT_TYPE_TO_NAME[quantization_type]}-gs{group_size}"
65
  else:
66
+ repo_name = f"{username}/{model_name.split('/')[-1]}-ao-{MAP_QUANT_TYPE_TO_NAME[quantization_type]}"
67
  if repo_name in model_names:
68
  return f"Model '{repo_name}' already exists in your repository."
69
  else: