Upload UltravoxPipeline

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f7dac54d53df3d158e6024aec3d59c3ded602e25492c210c276be98e8ec71e39
+size 117473752

ultravox_config.py

@@ -139,3 +139,6 @@ class UltravoxConfig(transformers.PretrainedConfig):
         self.initializer_range = self.text_config.initializer_range
 
         super().__init__(**kwargs)
+
+
+UltravoxConfig.register_for_auto_class()

ultravox_model.py

@@ -397,8 +397,6 @@ class UltravoxProjector(nn.Sequential):
         return hidden_states
 
 
-transformers.AutoModelForCausalLM.register(
-    ultravox_config.UltravoxConfig, UltravoxModel
-)
+UltravoxModel.register_for_auto_class("AutoModelForCausalLM")
 
 transformers.activations.ACT2FN["swiglu"] = SwiGLU

ultravox_pipeline.py

@@ -101,9 +101,8 @@ class UltravoxPipeline(transformers.Pipeline):
         return output_text
 
 
-transformers.pipeline
 transformers.pipelines.PIPELINE_REGISTRY.register_pipeline(
-    "ultravox-pipeline",  # TODO: make it broader later on
+    "ultravox-pipeline",
     pipeline_class=UltravoxPipeline,
     pt_model=ultravox_model.UltravoxModel,
     default={"pt": ("fixie-ai/ultravox-v0.2", "main")},