Update finetune_sample/lora_test.py
Browse files
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 = [
|