fix bug

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

src/submission/check_validity.py

@@ -154,8 +154,9 @@ def get_quantized_model_parameters_memory(model_info: ModelInfo, quant_method=""
         size_gb = round(meta.size / 1e9, 2)
         return params_b, size_gb
 
-    if "pytorch_model.bin.index.json" in filenames:
-        index_path = hf_hub_download(model_info.id, filename="pytorch_model.bin.index.json")
+    if "pytorch_model.bin.index.json" in filenames or "model.safetensors.index.json" in filenames:
+        json_file_name = "pytorch_model.bin.index.json" if "pytorch_model.bin.index.json" in filenames else "model.safetensors.index.json"
+        index_path = hf_hub_download(model_info.id, filename=json_file_name)
         """
         {
         "metadata": {