Skip to content

Commit

Permalink
Apply isort and black reformatting
Browse files Browse the repository at this point in the history
Signed-off-by: akoumpa <[email protected]>
  • Loading branch information
akoumpa committed Jan 20, 2025
1 parent 9d62688 commit 830a84f
Showing 1 changed file with 9 additions and 6 deletions.
15 changes: 9 additions & 6 deletions nemo/collections/llm/peft/lora.py
Original file line number Diff line number Diff line change
Expand Up @@ -82,12 +82,15 @@ def __init__(
if orig_linear.bias is not None:
self.bias.data.copy_(orig_linear.bias.data)
# initialize the adapte
LinearAdapter._init_adapter(self, dim=dim,
alpha=alpha,
dropout=dropout,
dropout_position=dropout_position,
lora_A_init_method=lora_A_init_method,
lora_dtype=lora_dtype)
LinearAdapter._init_adapter(
self,
dim=dim,
alpha=alpha,
dropout=dropout,
dropout_position=dropout_position,
lora_A_init_method=lora_A_init_method,
lora_dtype=lora_dtype,
)

@staticmethod
def _init_adapter(
Expand Down

0 comments on commit 830a84f

Please sign in to comment.