Add AutoProcessor registration (#102)

fixie-ai · Aug 28, 2024 · 84fc928 · 84fc928
1 parent 44eebcc
commit 84fc928
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/ultravox/model/ultravox_processing.py b/ultravox/model/ultravox_processing.py
@@ -205,4 +205,6 @@ def model_input_names(self):
         return list(set(tokenizer_input_names + audio_processor_input_names))
 
 
+UltravoxProcessor.register_for_auto_class()
+
 transformers.AutoProcessor.register(UltravoxConfig, UltravoxProcessor)