sitammeur commited on
Commit
e32e347
·
verified ·
1 Parent(s): 052841c

Update src/app/model.py

Browse files
Files changed (1) hide show
  1. 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