AlexHung29629 commited on
Commit
5e2c760
·
verified ·
1 Parent(s): ec22167

Update modeling_llama3.py

Browse files
Files changed (1) hide show
  1. modeling_llama3.py +1 -0
modeling_llama3.py CHANGED
@@ -421,6 +421,7 @@ class Llama3ForConditionalGeneration(MllamaPreTrainedModel, GenerationMixin):
421
 
422
  self.vision_model = MllamaVisionModel._from_config(config.vision_config)
423
  self.language_model = MllamaForCausalLM._from_config(config.text_config)
 
424
  self.embed_tokens = Llama3Embedding(config.audio_config, config.text_config)
425
  self.multi_modal_projector = nn.Linear(
426
  config.vision_config.vision_output_dim,
 
421
 
422
  self.vision_model = MllamaVisionModel._from_config(config.vision_config)
423
  self.language_model = MllamaForCausalLM._from_config(config.text_config)
424
+ self.language_model.get_input_embeddings().weight.required_grad = False
425
  self.embed_tokens = Llama3Embedding(config.audio_config, config.text_config)
426
  self.multi_modal_projector = nn.Linear(
427
  config.vision_config.vision_output_dim,