andstor commited on
Commit
007d5b8
·
verified ·
1 Parent(s): 011e203

Update src/model_utils.py

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