Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
Signed-off-by: Alexandros Koumparoulis <[email protected]>
  • Loading branch information
akoumpa committed Oct 23, 2024
1 parent 1ba27a8 commit 15d13e8
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion nemo/collections/llm/peft/lora.py
Original file line number Diff line number Diff line change
Expand Up @@ -73,7 +73,7 @@ def forward(self, x):

class LinearAdapter(nn.Module):
def __init__(self, orig_linear, dim=8, alpha=32, dropout=0.1, dropout_position='post', lora_A_init_method='xavier'):
super(LoraLinear, self).__init__()
super(LinearAdapter, self).__init__()
assert isinstance(orig_linear, nn.Linear)

self.orig_linear = orig_linear
Expand Down

0 comments on commit 15d13e8

Please sign in to comment.