Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Update src/submission/check_validity.py
Browse files
src/submission/check_validity.py
CHANGED
@@ -64,7 +64,7 @@ def get_model_size(model_info: ModelInfo, precision: str):
|
|
64 |
try:
|
65 |
model_size = round(model_info.safetensors["total"] / 1e9, 3)
|
66 |
except (AttributeError, TypeError):
|
67 |
-
return
|
68 |
|
69 |
size_factor = 8 if (precision == "GPTQ" or "gptq" in model_info.modelId.lower()) else 1
|
70 |
model_size = size_factor * model_size
|
|
|
64 |
try:
|
65 |
model_size = round(model_info.safetensors["total"] / 1e9, 3)
|
66 |
except (AttributeError, TypeError):
|
67 |
+
return 0 # Unknown model sizes are indicated as 0, see NUMERIC_INTERVALS in app.py
|
68 |
|
69 |
size_factor = 8 if (precision == "GPTQ" or "gptq" in model_info.modelId.lower()) else 1
|
70 |
model_size = size_factor * model_size
|