Wisdom Chen commited on
Commit
e7e84ea
·
unverified ·
1 Parent(s): 2566e6f

Update model.py

Browse files

Add the "use_fast=False"

Files changed (1) hide show
  1. model.py +2 -1
model.py CHANGED
@@ -84,7 +84,8 @@ def initialize_models() -> bool:
84
  llm_tokenizer = AutoTokenizer.from_pretrained(
85
  model_name,
86
  token=hf_token,
87
- trust_remote_code=True
 
88
  )
89
  llm_tokenizer.pad_token = llm_tokenizer.eos_token
90
 
 
84
  llm_tokenizer = AutoTokenizer.from_pretrained(
85
  model_name,
86
  token=hf_token,
87
+ trust_remote_code=True,
88
+ use_fast=False
89
  )
90
  llm_tokenizer.pad_token = llm_tokenizer.eos_token
91