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:
|