Update modeling_llama3.py
Browse files- 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,
|