Upload modeling_backbone_components.py with huggingface_hub

modeling_backbone_components.py

@@ -14,15 +14,26 @@ from transformers.modeling_outputs import BaseModelOutputWithPast
 from transformers.modeling_rope_utils import ROPE_INIT_FUNCTIONS, dynamic_rope_update
 from transformers.utils import logging
 
-from configuration_mimi import MimiConfig
-from modeling_mimi_clean import (
-    MimiAttention, 
-    MimiMLP, 
-    MimiLayerScale,
-    MimiRotaryEmbedding,
-    apply_rotary_pos_emb,
-    MIMI_ATTENTION_CLASSES
-)
+try:
+    from .configuration_mimi import MimiConfig
+    from .modeling_mimi_clean import (
+        MimiAttention, 
+        MimiMLP, 
+        MimiLayerScale,
+        MimiRotaryEmbedding,
+        apply_rotary_pos_emb,
+        MIMI_ATTENTION_CLASSES
+    )
+except ImportError:
+    from configuration_mimi import MimiConfig
+    from modeling_mimi_clean import (
+        MimiAttention, 
+        MimiMLP, 
+        MimiLayerScale,
+        MimiRotaryEmbedding,
+        apply_rotary_pos_emb,
+        MIMI_ATTENTION_CLASSES
+    )
 
 logger = logging.get_logger(__name__)