Update modeling_intern_vit.py

modeling_intern_vit.py

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