Update mllama_audio_model.py

mllama_audio_model.py

@@ -12,7 +12,7 @@ class Llama3Embedding(MllamaPreTrainedModel):
     base_model_prefix = "audio_model"
     def __init__(self, config: Llama3Config):
         super().__init__(config)
-        assert config.audio_config.output_hidden_size * 2 == config.text_config.hidden_size
+        #assert config.audio_config.output_hidden_size * 2 == config.text_config.hidden_size
         self.text_embeddings = nn.Embedding(config.text_config.vocab_size, config.text_config.hidden_size, config.text_config.pad_token_id)
         assert config.audio_config.add_adapter == True
         self.audio_model = Wav2Vec2Model(config.audio_config)