Upload modeling_text_sync_mimi.py with huggingface_hub

modeling_text_sync_mimi.py

@@ -6,6 +6,7 @@ from typing import Optional, Dict, List, Union
 
 try:
     from .configuration_mimi import MimiConfig
+    from .configuration_text_sync_mimi import TextSyncMimiConfig
     from .modeling_mimi_clean import MimiPreTrainedModel, MimiModel
     from .modeling_backbone_components import (
         CrossAttentionTransformer,
@@ -13,6 +14,7 @@ try:
     )
 except ImportError:
     from configuration_mimi import MimiConfig
+    from configuration_text_sync_mimi import TextSyncMimiConfig
     from modeling_mimi_clean import MimiPreTrainedModel, MimiModel
     from modeling_backbone_components import (
         CrossAttentionTransformer,
@@ -41,6 +43,8 @@ class TextSyncMimi(MimiPreTrainedModel):
     - End Token Classifier: Predicts when to stop generating
     """
     
+    config_class = TextSyncMimiConfig
+    
     def __init__(
         self, 
         config: Optional[Union[MimiConfig, 'TextSyncMimiConfig']] = None,