Update src/model_utils.py
Browse files- 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:
|