AlexHung29629
/

test_mllama_v14

Feature Extraction

Model card Files Files and versions Community

AlexHung29629 commited on Nov 20, 2024

Commit

508fc91

·

verified ·

1 Parent(s): 41dd4ce

Update ultravox_model.py

Files changed (1) hide show

ultravox_model.py +1 -1

ultravox_model.py CHANGED Viewed

@@ -304,7 +304,7 @@ class UltravoxModel(transformers.LlamaPreTrainedModel):
         if isinstance(
             audio_tower,
-            (transformers.Wav2Vec2BertModel, transformers.WhisperModel),
         ):
             # For these models we only need the encoder part
             # Wav2Vec2BertModel -> Wav2Vec2BertEncoder

         if isinstance(
             audio_tower,
+            (transformers.WhisperModel),
         ):
             # For these models we only need the encoder part
             # Wav2Vec2BertModel -> Wav2Vec2BertEncoder