OpenGVLab
/

InternVL-14B-224px

Image Feature Extraction

feature-extraction

Model card Files Files and versions Community

czczup commited on Feb 3

Commit

964aedc

•

1 Parent(s): 0d5e783

Update modeling_intern_vit.py

Files changed (1) hide show

modeling_intern_vit.py +1 -0

modeling_intern_vit.py CHANGED Viewed

@@ -279,6 +279,7 @@ class InternVisionEncoder(nn.Module):
 class InternVisionModel(PreTrainedModel):
     main_input_name = 'pixel_values'
     config_class = InternVisionConfig
     def __init__(self, config: InternVisionConfig):
         super().__init__(config)

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