fix style

1347027216 · Sep 5, 2023 · 5c3e0e2 · 5c3e0e2
1 parent 019b04a
commit 5c3e0e2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/scripts/training/peft/peft_model.py b/scripts/training/peft/peft_model.py
@@ -288,7 +288,7 @@ def __getattr__(self, name: str):
         except AttributeError:
             return getattr(self.base_model, name)
 
-    def forward(self, *args, **kwargs):
+    def forward(self, *args, **kwargs):  # pylint: disable=E0202
         """
         Forward pass of the model.
         """
@@ -355,7 +355,7 @@ def __init__(self, model, peft_config: PeftConfig):
         # to make sure classifier layer is trainable
         _set_trainable(self)
 
-    def forward(
+    def forward(    # pylint: disable=W0221
         self,
         input_ids=None,
         attention_mask=None,

diff --git a/scripts/training/peft/tuners/lora.py b/scripts/training/peft/tuners/lora.py
@@ -381,7 +381,7 @@ def forward(self, x: torch.Tensor):
 
 class MergedLinear(nn.Linear, LoraLayer):
     # Lora implemented in a dense layer
-    def __init__(
+    def __init__(    # pylint: disable=W0102
         self,
         in_features: int,
         out_features: int,