Bram Vanroy commited on
Commit
1c70b16
1 Parent(s): 9ffaef5

add trust_remote_code

Browse files
Files changed (1) hide show
  1. generate_overview_json.py +1 -1
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():