fixie-ai
/

ultravox-v0_2

Feature Extraction

Model card Files Files and versions Community

farzadab commited on Jul 25

Commit

d85845b

•

1 Parent(s): ffd36f6

Update ultravox_model.py

Files changed (1) hide show

ultravox_model.py +1 -0

ultravox_model.py CHANGED Viewed

@@ -33,6 +33,7 @@ class UltravoxModel(transformers.LlamaPreTrainedModel):
     config_class = UltravoxConfig
     config: UltravoxConfig  # for type hinting
     _no_split_modules = ["Wav2Vec2Model", "WhisperEncoder", "LlamaDecoderLayer"]
     def __init__(self, config: UltravoxConfig):
         super().__init__(config)

     config_class = UltravoxConfig
     config: UltravoxConfig  # for type hinting
     _no_split_modules = ["Wav2Vec2Model", "WhisperEncoder", "LlamaDecoderLayer"]
+    _keys_to_ignore_on_load_missing = ["audio_tower"]
     def __init__(self, config: UltravoxConfig):
         super().__init__(config)