Wisdom Chen commited on
Commit
c9e87c4
·
unverified ·
1 Parent(s): 501b042

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +4 -5
model.py CHANGED
@@ -88,21 +88,20 @@ def initialize_models() -> bool:
88
  bnb_4bit_quant_type="nf4"
89
  )
90
 
91
- # Initialize tokenizer with updated parameters based on documentation
92
  llm_tokenizer = AutoTokenizer.from_pretrained(
93
  model_name,
94
- use_fast=True, # Add this line
95
- padding_side="left", # Add this line
96
  trust_remote_code=True
97
  )
98
  llm_tokenizer.pad_token = llm_tokenizer.eos_token
99
 
100
- # Initialize model with updated parameters
101
  llm_model = AutoModelForCausalLM.from_pretrained(
102
  model_name,
 
 
103
  device_map="auto",
104
  torch_dtype=torch.float16,
105
- quantization_config=quantization_config,
106
  trust_remote_code=True
107
  )
108
  llm_model.eval()
 
88
  bnb_4bit_quant_type="nf4"
89
  )
90
 
91
+ # Initialize tokenizer with specific version requirements
92
  llm_tokenizer = AutoTokenizer.from_pretrained(
93
  model_name,
94
+ use_auth_token=hf_token, # Changed from token to use_auth_token
 
95
  trust_remote_code=True
96
  )
97
  llm_tokenizer.pad_token = llm_tokenizer.eos_token
98
 
 
99
  llm_model = AutoModelForCausalLM.from_pretrained(
100
  model_name,
101
+ use_auth_token=hf_token, # Changed from token to use_auth_token
102
+ quantization_config=quantization_config,
103
  device_map="auto",
104
  torch_dtype=torch.float16,
 
105
  trust_remote_code=True
106
  )
107
  llm_model.eval()