fix bug

Signed-off-by: n1ck-guo <heng.guo@intel.com>

src/submission/check_validity.py

@@ -147,8 +147,9 @@ def get_quantized_model_parameters_memory(model_info: ModelInfo, quant_method=""
         print(str(e))
 
     filenames = [sib.rfilename for sib in model_info.siblings]
-    if "pytorch_model.bin" in filenames:
-        url = hf_hub_url(model_info.id, filename="pytorch_model.bin")
+    if "pytorch_model.bin" in filenames or "model.safetensors" in filenames:
+        bin_filename = "pytorch_model.bin" if "pytorch_model.bin" in filenames else "model.safetensors"
+        url = hf_hub_url(model_info.id, filename=bin_filename)
         meta = get_hf_file_metadata(url)
         params_b = round(meta.size * 2 / 1e9, 2)
         size_gb = round(meta.size / 1e9, 2)