Bram Vanroy
commited on
Commit
•
1c70b16
1
Parent(s):
9ffaef5
add trust_remote_code
Browse files
generate_overview_json.py
CHANGED
@@ -11,7 +11,7 @@ def get_num_parameters(model_name: str) -> int:
|
|
11 |
info = model_info(model_name)
|
12 |
return info.safetensors["total"]
|
13 |
except Exception:
|
14 |
-
return AutoModelForCausalLM.from_pretrained(model_name).num_parameters()
|
15 |
|
16 |
|
17 |
def main():
|
|
|
11 |
info = model_info(model_name)
|
12 |
return info.safetensors["total"]
|
13 |
except Exception:
|
14 |
+
return AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True).num_parameters()
|
15 |
|
16 |
|
17 |
def main():
|