update data collection script

generate_overview_json.py

@@ -33,16 +33,18 @@ def main():
         if "pretrained" not in model_args:
             continue
 
-        results[short_name]["model_name"] = model_args["pretrained"]
-        results[short_name]["compute_dtype"] = model_args.get("dtype", None)
-        results[short_name]["quantization"] = None
+        results[short_name] = {
+            "model_name": model_args["pretrained"],
+            "compute_dtype": model_args.get("dtype", None),
+            "quantization": None,
+            "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:
             results[short_name]["quantization"] = "4-bit"
 
-        results[short_name]["num_parameters"] = get_num_parameters(model_args["pretrained"])
-
     pprint(results)
     pf_overview.write_text(json.dumps(results, indent=4), encoding="utf-8")