GMFTBY commited on
Commit
9522941
1 Parent(s): e592474

delete the useless auth token

Browse files
Files changed (1) hide show
  1. model/openllama.py +1 -1
model/openllama.py CHANGED
@@ -105,7 +105,7 @@ class OpenLLAMAPEFTModel(nn.Module):
105
 
106
  with init_empty_weights():
107
  config = LlamaConfig.from_pretrained(vicuna_ckpt_path, use_auth_token=os.environ['API_TOKEN'])
108
- self.llama_model = LlamaForCausalLM(config, use_auth_token=os.environ['API_TOKEN'])
109
 
110
  self.llama_model = load_checkpoint_and_dispatch(self.llama_model, vicuna_ckpt_path, device_map='sequential')
111
 
 
105
 
106
  with init_empty_weights():
107
  config = LlamaConfig.from_pretrained(vicuna_ckpt_path, use_auth_token=os.environ['API_TOKEN'])
108
+ self.llama_model = LlamaForCausalLM(config)
109
 
110
  self.llama_model = load_checkpoint_and_dispatch(self.llama_model, vicuna_ckpt_path, device_map='sequential')
111