andstor commited on
Commit
47d7bc7
·
verified ·
1 Parent(s): 09c6949

Update src/model_utils.py

Browse files
Files changed (1) hide show
  1. src/model_utils.py +1 -0
src/model_utils.py CHANGED
@@ -43,6 +43,7 @@ def get_model(model_name: str, library: str, access_token: str):
43
  if library == "auto":
44
  library = None
45
  model_name = extract_from_url(model_name)
 
46
  try:
47
  model = create_empty_model(model_name, library_name=library, trust_remote_code=True, access_token=access_token)
48
  except GatedRepoError:
 
43
  if library == "auto":
44
  library = None
45
  model_name = extract_from_url(model_name)
46
+ access_token = access_token or None
47
  try:
48
  model = create_empty_model(model_name, library_name=library, trust_remote_code=True, access_token=access_token)
49
  except GatedRepoError: