Update multipurpose_chatbot/engines/modeling_sealmm.py

multipurpose_chatbot/engines/modeling_sealmm.py

@@ -597,7 +597,7 @@ class SeaLMMForCausalLM(LlavaPreTrainedModel):
         # self.vision_tower = vision_tower or LlavaCLIPVisionModel(config=config.vision_config)
         self.vision_tower = vision_tower or SeaLMMCLIPVisionModel(config=config.vision_config)
         self.multi_modal_projector = SeaLMMMultiModalProjector(config)
-        self.vocab_size = config.vocab_size
+        self.vocab_size = config.text_config.vocab_size
         self.language_model = language_model or AutoModelForCausalLM.from_config(
             config.text_config, attn_implementation=config._attn_implementation
         )