fix: import & logic error

modeling_lolicore.py

@@ -45,7 +45,7 @@ from transformers.utils import (
     logging,
     replace_return_docstrings,
 )
-from configuration_lolicore import LoliCoreConfig
+from .configuration_lolicore import LoliCoreConfig
 
 
 if is_flash_attn_2_available():
@@ -335,7 +335,7 @@ class LoliCoreJump(nn.Module):
         self.intermediate_size = intermediate_size
 
     def forward(self, hidden_states):
-        return hidden_states
+        return torch.zeros_like(hidden_states)
         
 
 class LoliCoreMoEMLP(nn.Module):