make

huggingface · Dec 17, 2024 · 407cd3e · 407cd3e
1 parent 2da255d
commit 407cd3e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 8 deletions.
diff --git a/src/transformers/__init__.py b/src/transformers/__init__.py
@@ -463,7 +463,6 @@
     "models.gpt_bigcode": ["GPTBigCodeConfig"],
     "models.gpt_neo": ["GPTNeoConfig"],
     "models.gpt_neox": ["GPTNeoXConfig"],
-    "models.moonshine": ["MoonshineConfig"],
     "models.gpt_neox_japanese": ["GPTNeoXJapaneseConfig"],
     "models.gpt_sw3": [],
     "models.gptj": ["GPTJConfig"],
@@ -5792,7 +5791,6 @@
         WhisperProcessor,
         WhisperTokenizer,
     )
-    from .models.moonshine import MoonshineConfig
     from .models.x_clip import (
         XCLIPConfig,
         XCLIPProcessor,
@@ -7466,6 +7464,7 @@
         )
         from .models.moonshine import (
             MoonshineForCausalLM,
+            MoonshineForConditionalGeneration,
             MoonshineForQuestionAnswering,
             MoonshineForSequenceClassification,
             MoonshineForTokenClassification,
@@ -8172,11 +8171,6 @@
             WhisperModel,
             WhisperPreTrainedModel,
         )
-        from .models.moonshine import (
-            MoonshineModel, 
-            MoonshineForConditionalGeneration,
-            MoonshinePreTrainedModel,
-        )
         from .models.x_clip import (
             XCLIPModel,
             XCLIPPreTrainedModel,

diff --git a/src/transformers/models/auto/modeling_auto.py b/src/transformers/models/auto/modeling_auto.py
@@ -908,14 +908,14 @@
 
 MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPING_NAMES = OrderedDict(
     [
+        ("moonshine", "MoonshineForConditionalGeneration"),
         ("pop2piano", "Pop2PianoForConditionalGeneration"),
         ("seamless_m4t", "SeamlessM4TForSpeechToText"),
         ("seamless_m4t_v2", "SeamlessM4Tv2ForSpeechToText"),
         ("speech-encoder-decoder", "SpeechEncoderDecoderModel"),
         ("speech_to_text", "Speech2TextForConditionalGeneration"),
         ("speecht5", "SpeechT5ForSpeechToText"),
         ("whisper", "WhisperForConditionalGeneration"),
-        ("moonshine", "MoonshineForConditionalGeneration"),
     ]
 )