fix duplicate-code warnings

2024-12-23 14:22:33 -05:00
parent 5b90da0be3
commit a3fd5074a9
1 changed files with 2 additions and 0 deletions
--- a/src/axolotl/integrations/diff_transformer/modeling_diff_attn.py
+++ b/src/axolotl/integrations/diff_transformer/modeling_diff_attn.py
@@ -96,6 +96,7 @@ class LlamaDifferentialModel(LlamaDifferentialPreTrainedModel):
        )
        self.norm = LlamaRMSNorm(config.hidden_size, eps=config.rms_norm_eps)

+    # pylint: disable=duplicate-code
    def forward(
        self,
        input_ids: Optional[torch.LongTensor] = None,
@@ -322,6 +323,7 @@ class LlamaDifferentialDecoderLayer(nn.Module):
            config.hidden_size, eps=config.rms_norm_eps
        )

+    # pylint: disable=duplicate-code
    def forward(
        self,
        hidden_states: torch.Tensor,