use model_info because Omar tells me to :-(

generate_overview_json.py

@@ -1,15 +1,15 @@
 import json
 from pathlib import Path
 
-from huggingface_hub import get_safetensors_metadata
+from huggingface_hub import model_info
 from tqdm import tqdm
 from transformers import AutoModelForCausalLM
 
 
 def get_num_parameters(model_name: str) -> int:
     try:
-        metadata = get_safetensors_metadata(model_name)
-        return sum(metadata.parameter_count.values())
+        info = model_info(model_name)
+        return info.safetensors["total"]
     except Exception:
         return AutoModelForCausalLM.from_pretrained(model_name).num_parameters()