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

Merge pull request #9 from huggingface/smangrul/add-examples-fixes-docs

Browse Source 
fixes
This commit is contained in:
Sourab Mangrulkar
2022-12-05 17:56:06 +05:30
committed by GitHub
parent 85053f0145 9c8e33df6f
commit b55ea93033
2 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/pet/pet_model.py
+2 -2
View File
@@ -178,7 +178,7 @@ class PETModelForSequenceClassification(PETModel):
break
# to make sure classifier layer is trainable
_set_trainable(self.base_model)
_set_trainable(self)
def forward(
self,
@@ -540,7 +540,7 @@ class PETModelForTokenClassification(PETModel):
break
# to make sure classifier layer is trainable
_set_trainable(self.base_model)
_set_trainable(self)
def forward(
self,
src/pet/utils/other.py
-2
View File
@@ -42,8 +42,6 @@ def _set_trainable(model):
for name, param in model.named_parameters():
if any(module_name in name for module_name in model.modules_to_save):
param.requires_grad = True
else:
param.requires_grad = False
def fsdp_auto_wrap_policy(model):