From 7d9ec2c77ecb465cd76bb95a99b1651b594385e0 Mon Sep 17 00:00:00 2001
From: Dan Saunders <dan@axolotl.ai>
Date: Mon, 23 Dec 2024 14:22:33 -0500
Subject: [PATCH] fix duplicate-code warnings

---
 src/axolotl/integrations/diff_transformer/modeling_diff_attn.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/axolotl/integrations/diff_transformer/modeling_diff_attn.py b/src/axolotl/integrations/diff_transformer/modeling_diff_attn.py
index 594970716..4b97bfe10 100644
--- a/src/axolotl/integrations/diff_transformer/modeling_diff_attn.py
+++ b/src/axolotl/integrations/diff_transformer/modeling_diff_attn.py
@@ -96,6 +96,7 @@ def __init__(self, config: LlamaDifferentialConfig):
         )
         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 @@ def __init__(
             config.hidden_size, eps=config.rms_norm_eps
         )
 
+    # pylint: disable=duplicate-code
     def forward(
         self,
         hidden_states: torch.Tensor,