OpenGVLab
/

InternVL-Chat-V1-1

Visual Question Answering

feature-extraction

Model card Files Files and versions Community

czczup commited on Feb 4

Commit

ad5ccb7

•

1 Parent(s): 2743f6b

Update modeling_intern_vit.py

Files changed (1) hide show

modeling_intern_vit.py +1 -1

modeling_intern_vit.py CHANGED Viewed

@@ -349,7 +349,7 @@ class InternVisionEncoder(nn.Module):
 class InternVisionModel(PreTrainedModel):
     main_input_name = 'pixel_values'
     config_class = InternVisionConfig
-    _no_split_modules = ['InternAttention']
     def __init__(self, config: InternVisionConfig):
         super().__init__(config)

 class InternVisionModel(PreTrainedModel):
     main_input_name = 'pixel_values'
     config_class = InternVisionConfig
+    _no_split_modules = ['InternVisionEncoderLayer']
     def __init__(self, config: InternVisionConfig):
         super().__init__(config)