Spaces:
Sleeping
Sleeping
Commit
•
b37adc3
1
Parent(s):
3b554b5
Update src/submission/check_validity.py (#509)
Browse files- Update src/submission/check_validity.py (b9fea6f0e7cf6b2580c52554768dad9f1971a001)
Co-authored-by: Sean Cho <BearSean@users.noreply.huggingface.co>
src/submission/check_validity.py
CHANGED
@@ -64,7 +64,7 @@ def is_model_on_hub(model_name: str, revision: str, token: str = None, trust_rem
|
|
64 |
|
65 |
|
66 |
def get_model_size(model_info: ModelInfo, precision: str):
|
67 |
-
size_pattern = size_pattern = re.compile(r"(\d
|
68 |
try:
|
69 |
model_size = round(model_info.safetensors["total"] / 1e9, 3)
|
70 |
except (AttributeError, TypeError ):
|
|
|
64 |
|
65 |
|
66 |
def get_model_size(model_info: ModelInfo, precision: str):
|
67 |
+
size_pattern = size_pattern = re.compile(r"(\d+\.)?\d+(b|m)")
|
68 |
try:
|
69 |
model_size = round(model_info.safetensors["total"] / 1e9, 3)
|
70 |
except (AttributeError, TypeError ):
|