Wisdom Chen commited on
Commit
4122243
·
unverified ·
1 Parent(s): d2a75a6

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +2 -3
model.py CHANGED
@@ -76,7 +76,7 @@ def initialize_models() -> bool:
76
  )
77
 
78
  # Get token from Streamlit secrets
79
- hf_token = st.secrets.get("HUGGINGFACE_TOKEN")
80
  if not hf_token:
81
  raise ValueError("HUGGINGFACE_TOKEN not found in Streamlit secrets")
82
 
@@ -84,8 +84,7 @@ def initialize_models() -> bool:
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
 
 
76
  )
77
 
78
  # Get token from Streamlit secrets
79
+ hf_token = st.secrets["HUGGINGFACE_TOKEN"]
80
  if not hf_token:
81
  raise ValueError("HUGGINGFACE_TOKEN not found in Streamlit secrets")
82
 
 
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