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

update README and fix token_cls example

Browse Source
This commit is contained in:
Sourab Mangrulkar
2023-02-08 18:54:46 +05:30
parent 73cd16b7b5
commit 8083c9515f
2 changed files with 11 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/token_classification/peft_lora_token_cls.ipynb
+3 -13
View File
@@ -827,7 +827,7 @@
}
],
"source": [
"from peft import get_peft_config, LoraModel, get_peft_model, LoraConfig, TaskType\n",
"from peft import get_peft_config, PeftModel, get_peft_model, LoraConfig, TaskType\n",
"peft_config = LoraConfig(\n",
" task_type=TaskType.TOKEN_CLS,\n",
" inference_mode=False,\n",
@@ -1070,17 +1070,7 @@
"metadata": {},
"outputs": [],
"source": [
"from peft import get_peft_model_state_dict\n",
"to_return = get_peft_model_state_dict(model)\n"
]
},
{
"cell_type": "code",
"execution_count": 20,
"metadata": {},
"outputs": [],
"source": [
"torch.save(to_return, \"layoutlm_funsd.pt\")"
"model.save_pretrained(\"peft_layoutlm\")\n"
]
},
{
@@ -1097,7 +1087,7 @@
}
],
"source": [
"!du -h \"layoutlm_funsd.pt\""
"!du -h \"peft_layoutlm/adapter_model.bin\""
]
},
{