Skip to content
This repository was archived by the owner on Dec 14, 2023. It is now read-only.

Commit 28c8f67

Browse files
Remove redundant variable in module unfreeze
1 parent fc6b827 commit 28c8f67

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

train.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -367,9 +367,7 @@ def handle_trainable_modules(model, trainable_modules=None, is_enabled=True, neg
367367
model.requires_grad_(is_enabled)
368368
unfrozen_params =len(list(model.parameters()))
369369
break
370-
should_negate = negate_params(name, negation)
371-
if should_negate: continue
372-
370+
373371
if tm in name and 'lora' not in name:
374372
for m in module.parameters():
375373
m.requires_grad_(is_enabled)

0 commit comments

Comments
 (0)