Update modeling_custom_seq2seq_llm.py

modeling_custom_seq2seq_llm.py

@@ -3,7 +3,7 @@ import torch.nn as nn
 from torch.nn import CrossEntropyLoss
 from transformers.modeling_outputs import Seq2SeqLMOutput
 from transformers.activations import ACT2FN
-from .flash_atten import MHA  # Import the MHA class from the provided implementation
+from flash_atten import MHA  # Import the MHA class from the provided implementation
 from liger_kernel.transformers.cross_entropy import LigerCrossEntropyLoss
 from liger_kernel.transformers.rms_norm import LigerRMSNorm
 from liger_kernel.transformers.swiglu import LigerSwiGLUMLP