choco9966 commited on
Commit
5b935ed
1 Parent(s): 10c5abc

[Fix] trust_remote_code to False

Browse files
Files changed (1) hide show
  1. src/load_from_hub.py +1 -1
src/load_from_hub.py CHANGED
@@ -131,7 +131,7 @@ def get_evaluation_queue_df(
131
 
132
  def is_model_on_hub(model_name: str, revision: str) -> bool:
133
  try:
134
- AutoConfig.from_pretrained(model_name, revision=revision, trust_remote_code=True)
135
  return True, None
136
 
137
  except ValueError:
 
131
 
132
  def is_model_on_hub(model_name: str, revision: str) -> bool:
133
  try:
134
+ AutoConfig.from_pretrained(model_name, revision=revision, trust_remote_code=False)
135
  return True, None
136
 
137
  except ValueError: