Update model/openlamm.py
Browse files- 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 |
|