Skip to content

Commit

Permalink
Apply isort and black reformatting
Browse files Browse the repository at this point in the history
Signed-off-by: cuichenx <cuichenx@users.noreply.github.com>
  • Loading branch information
cuichenx committed Sep 26, 2024
1 parent 9d7b39c commit d1c2069
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions nemo/lightning/pytorch/callbacks/peft.py
Original file line number Diff line number Diff line change
Expand Up @@ -143,6 +143,7 @@ def apply_transform(self, trainer):
def adapter_key_filter(self, key: str) -> bool:
return key in self.trainable_params or ".adapter." in key or key.endswith(".adapters")


class AdapterWrapper(nn.Module):
"""Abstract base class for wrapping modules with adapters in Parameter-Efficient Fine-Tuning (PEFT).
Expand Down

0 comments on commit d1c2069

Please sign in to comment.