Adjusted validity for large models

src/submission/check_validity.py

@@ -71,8 +71,8 @@ def get_model_size(model_info: ModelInfo, precision: str):
 
     size_factor = 8 if (precision == "GPTQ" or "gptq" in model_info.modelId.lower()) else 1
     model_size = size_factor * model_size
-    if model_size > 9 and precision != 'float16':
-        raise ValueError("For larger models we only support running in float16")
+    if model_size > 9 and precision == 'bfloat16':
+        raise ValueError("For larger models we don't support running with bfloat16")
     
     return model_size