fix generate overview

generate_overview_json.py

@@ -16,8 +16,7 @@ def main():
     for pfin in evals_dir.rglob("*.json"):
         if pfin.stem == "models":
             continue
-        short_name = pfin.stem.split("_")[2].lower()
-
+        short_name = pfin.stem.split("_", 2)[2].lower()
         data = json.loads(pfin.read_text(encoding="utf-8"))
         if "config" not in data:
             continue
@@ -33,12 +32,12 @@ def main():
         results[short_name] = {
             "model_name": model_args["pretrained"],
             "compute_dtype": model_args.get("dtype", None),
-            "quantization": None
+            "quantization": None,
+            "num_parameters": results[short_name]["num_parameters"]
+            if short_name in results and "num_parameters" in results[short_name]
+            else get_num_parameters(model_args["pretrained"]),
         }
 
-        if "num_parameters" not in results[short_name]:
-            results[short_name]["num_parameters"] = get_num_parameters(model_args["pretrained"])
-
         if "load_in_8bit" in model_args:
             results[short_name]["quantization"] = "8-bit"
         elif "load_in_4bit" in model_args:
@@ -47,5 +46,5 @@ def main():
     pf_overview.write_text(json.dumps(results, indent=4, sort_keys=True), encoding="utf-8")
 
 
-if __name__ == '__main__':
-    main()
+if __name__ == "__main__":
+    main()