OpenGVLab
/

InternVL-Chat-V1-1

Image-Text-to-Text

feature-extraction

Model card Files Files and versions Community

czczup commited on Feb 4

Commit

4ebe12a

•

1 Parent(s): ad5ccb7

Update modeling_internvl_chat.py

Files changed (1) hide show

modeling_internvl_chat.py +1 -1

modeling_internvl_chat.py CHANGED Viewed

@@ -23,7 +23,7 @@ logger = logging.get_logger(__name__)
 class InternVLChatModel(PreTrainedModel):
     config_class = InternVLChatConfig
     main_input_name = 'pixel_values'
-    _no_split_modules = ['InternAttention', 'LlamaDecoderLayer', 'LlamaForCausalLM']
     def __init__(self, config: InternVLChatConfig, vision_model=None, language_model=None):
         super().__init__(config)

 class InternVLChatModel(PreTrainedModel):
     config_class = InternVLChatConfig
     main_input_name = 'pixel_values'
+    _no_split_modules = ['InternVisionEncoderLayer', 'LlamaDecoderLayer', 'LlamaForCausalLM']
     def __init__(self, config: InternVLChatConfig, vision_model=None, language_model=None):
         super().__init__(config)