Update mllama_audio_model.py

mllama_audio_model.py

@@ -15,7 +15,7 @@ class MllamaAudioModel(MllamaPreTrainedModel):
         super().__init__(config)
         assert config.add_adapter is True, f'{type(self).__name__} requires add adapter to be true.'
         #assert config.output_hidden_size == text_embedding.weight.shape[1], f'Output hidden size({config.output_hidden_size}) of audio model and text embedding({text_embedding.weight.shape[1]}) must match!'
-        asseert config.output_hidden_size == text_config.hidden_size
+        assert config.output_hidden_size == text_config.hidden_size
         self.text_embedding = nn.Embedding(text_config.vocab_size + 8, text_config.hidden_size, text_config.pad_token_id)
         self.audio_embedding = Wav2Vec2BertModel(config)
         self.start_of_audio = nn.Parameter(data=torch.mean(text_embedding.weight, dim=0).unsqueeze(0), requires_grad=True)