openlamm commited on
Commit
8478a70
·
1 Parent(s): f05f5e8

Update model/openlamm.py

Browse files
Files changed (1) hide show
  1. model/openlamm.py +1 -1
model/openlamm.py CHANGED
@@ -203,7 +203,7 @@ class LAMMPEFTModel(nn.Module):
203
  target_modules=self.args['lora_target_modules']
204
  )
205
 
206
- self.llama_model = LlamaForCausalLM.from_pretrained(vicuna_ckpt_path)
207
  self.llama_model = get_peft_model(self.llama_model, peft_config)
208
  self.llama_model.print_trainable_parameters()
209
 
 
203
  target_modules=self.args['lora_target_modules']
204
  )
205
 
206
+ self.llama_model = LlamaForCausalLM.from_pretrained(vicuna_ckpt_path, cache_dir='~/.cache/')
207
  self.llama_model = get_peft_model(self.llama_model, peft_config)
208
  self.llama_model.print_trainable_parameters()
209