wassname/peft
1
0
Fork 0
mirror of https://github.com/wassname/peft.git synced 2026-09-10 12:20:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

example

Browse Source
This commit is contained in:
QingruZhang
2023-03-01 21:55:31 +00:00
parent 1a3680d8a7
commit 35cd771c97
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/conditional_generation/peft_adalora_seq2seq.py
+2 -2
View File
@@ -26,7 +26,7 @@ batch_size = 8
# creating model
peft_config = AdaLoraConfig(
r=8, lora_alpha=32, lora_dropout=0.1
r=8, lora_alpha=32, lora_dropout=0.1,
task_type=TaskType.SEQ_2_SEQ_LM,
inference_mode=False
)
@@ -151,7 +151,7 @@ model.save_pretrained(peft_model_id)
ckpt = f"{peft_model_id}/adapter_model.bin"
get_ipython().system('du -h $ckpt')
# get_ipython().system('du -h $ckpt')
from peft import PeftModel, PeftConfig