diff --git a/unsloth/models/llama.py b/unsloth/models/llama.py
index a4a6527ff2..a4b6552160 100644
--- a/unsloth/models/llama.py
+++ b/unsloth/models/llama.py
@@ -1065,7 +1065,7 @@ class LlamaExtendedRotaryEmbedding(torch.nn.Module):
         config = None, # [TODO] Hack to pass in config - need to remove later
     ):
         super().__init__()
-        if config is not None: return # [TODO] Hack to pass in config - need to remove later
+        # if config is not None: return # [TODO] Hack to pass in config - need to remove later
         
         self.dim = dim
         self.max_position_embeddings = max_position_embeddings