Update src/submission/check_validity.py (#509)

- Update src/submission/check_validity.py (b9fea6f0e7cf6b2580c52554768dad9f1971a001)


Co-authored-by: Sean Cho <[email protected]>

src/submission/check_validity.py

@@ -64,7 +64,7 @@ def is_model_on_hub(model_name: str, revision: str, token: str = None, trust_rem
 
 
 def get_model_size(model_info: ModelInfo, precision: str):
-    size_pattern = size_pattern = re.compile(r"(\d\.)?\d+(b|m)")
+    size_pattern = size_pattern = re.compile(r"(\d+\.)?\d+(b|m)")
     try:
         model_size = round(model_info.safetensors["total"] / 1e9, 3)
     except (AttributeError, TypeError ):