Update ultravox_pipeline.py

ultravox_pipeline.py

@@ -4,14 +4,14 @@ from typing import Any, Dict, List, Optional
 import transformers
 
 # We must use relative import in this directory to allow uploading to HF Hub
-import . ultravox_model
-import . ultravox_processing
+from .ultravox_model import UltravoxModel
+from .ultravox_processing import UltravoxProcessor
 
 
 class UltravoxPipeline(transformers.Pipeline):
     def __init__(
         self,
-        model: ultravox_model.UltravoxModel,
+        model: UltravoxModel,
         tokenizer: Optional[transformers.PreTrainedTokenizerBase] = None,
         audio_processor: Optional[transformers.ProcessorMixin] = None,
         **kwargs
@@ -26,7 +26,7 @@ class UltravoxPipeline(transformers.Pipeline):
                 model.config.audio_model_id
             )
 
-        self.processor = ultravox_processing.UltravoxProcessor(
+        self.processor = UltravoxProcessor(
             audio_processor, tokenizer=tokenizer, stack_factor=model.config.stack_factor
         )
 
@@ -104,7 +104,5 @@ class UltravoxPipeline(transformers.Pipeline):
 transformers.pipelines.PIPELINE_REGISTRY.register_pipeline(
     "ultravox-pipeline",
     pipeline_class=UltravoxPipeline,
-    pt_model=ultravox_model.UltravoxModel,
-    default={"pt": ("fixie-ai/ultravox-v0.2", "main")},
     type="multimodal",
 )