From 407cd3ed515eaf5a98513487c7bf563295d67a6a Mon Sep 17 00:00:00 2001
From: Eustache Le Bihan <eulebihan@gmail.com>
Date: Tue, 17 Dec 2024 14:05:44 +0100
Subject: [PATCH] make

---
 src/transformers/__init__.py                  | 8 +-------
 src/transformers/models/auto/modeling_auto.py | 2 +-
 2 files changed, 2 insertions(+), 8 deletions(-)

diff --git a/src/transformers/__init__.py b/src/transformers/__init__.py
index 43547fef8c5e18..507454b0450931 100755
--- 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
index 3a1d33b0b6caf4..ab9d02d7d3f23a 100644
--- a/src/transformers/models/auto/modeling_auto.py
+++ b/src/transformers/models/auto/modeling_auto.py
@@ -908,6 +908,7 @@
 
 MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPING_NAMES = OrderedDict(
     [
+        ("moonshine", "MoonshineForConditionalGeneration"),
         ("pop2piano", "Pop2PianoForConditionalGeneration"),
         ("seamless_m4t", "SeamlessM4TForSpeechToText"),
         ("seamless_m4t_v2", "SeamlessM4Tv2ForSpeechToText"),
@@ -915,7 +916,6 @@
         ("speech_to_text", "Speech2TextForConditionalGeneration"),
         ("speecht5", "SpeechT5ForSpeechToText"),
         ("whisper", "WhisperForConditionalGeneration"),
-        ("moonshine", "MoonshineForConditionalGeneration"),
     ]
 )