imports

huggingface · Dec 17, 2024 · b1f0909 · b1f0909
1 parent 0954133
commit b1f0909
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/transformers/__init__.py b/src/transformers/__init__.py
@@ -842,6 +842,7 @@
         "WhisperProcessor",
         "WhisperTokenizer",
     ],
+    "models.moonshine": ["MoonshineConfig"],
     "models.x_clip": [
         "XCLIPConfig",
         "XCLIPProcessor",
@@ -3722,6 +3723,13 @@
             "WhisperPreTrainedModel",
         ]
     )
+    _import_structure["models.moonshine"].extend(
+        [
+            "MoonshineForConditionalGeneration",
+            "MoonshineModel",
+            "MoonshinePreTrainedModel",
+        ]
+    )
     _import_structure["models.x_clip"].extend(
         [
             "XCLIPModel",
@@ -5784,6 +5792,7 @@
         WhisperProcessor,
         WhisperTokenizer,
     )
+    from .models.moonshine import MoonshineConfig
     from .models.x_clip import (
         XCLIPConfig,
         XCLIPProcessor,
@@ -8163,6 +8172,11 @@
             WhisperModel,
             WhisperPreTrainedModel,
         )
+        from .models.moonshine import (
+            MoonshineModel, 
+            MoonshineForConditionalGeneration,
+            MoonshinePreTrainedModel,
+        )
         from .models.x_clip import (
             XCLIPModel,
             XCLIPPreTrainedModel,