Spaces:
Runtime error
Runtime error
Update src/app/model.py
Browse files- src/app/model.py +1 -1
src/app/model.py
CHANGED
@@ -40,7 +40,7 @@ def load_model_and_tokenizer(model_name: str, device: str) -> Any:
|
|
40 |
token=access_token
|
41 |
)
|
42 |
model = model.to(device=device)
|
43 |
-
tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
|
44 |
model.eval()
|
45 |
|
46 |
# Log the successful loading of the model and tokenizer
|
|
|
40 |
token=access_token
|
41 |
)
|
42 |
model = model.to(device=device)
|
43 |
+
tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True, token=access_token)
|
44 |
model.eval()
|
45 |
|
46 |
# Log the successful loading of the model and tokenizer
|