andstor commited on
Commit
32462ec
·
verified ·
1 Parent(s): 03f4e17

Update src/model_utils.py

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