dahara1 commited on
Commit
15a5518
1 Parent(s): 963fdc9

Update finetune_sample/lora_test.py

Browse files
Files changed (1) hide show
  1. finetune_sample/lora_test.py +1 -1
finetune_sample/lora_test.py CHANGED
@@ -18,7 +18,7 @@ model = AutoModelForCausalLM.from_pretrained(
18
  quantization_config= GPTQConfig(bits=4, disable_exllama=True)
19
  )
20
 
21
- adapters_name = f'checkpoint-700}'
22
  p_model = PeftModel.from_pretrained(model, adapters_name)
23
 
24
  prompt_text = [
 
18
  quantization_config= GPTQConfig(bits=4, disable_exllama=True)
19
  )
20
 
21
+ adapters_name = f'checkpoint-700'
22
  p_model = PeftModel.from_pretrained(model, adapters_name)
23
 
24
  prompt_text = [