Spaces:
Build error
Build error
Wisdom Chen
commited on
Update model.py
Browse filesAdd the "use_fast=False"
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 |
|