Update modeling_intern_vit.py
Browse files- modeling_intern_vit.py +1 -0
modeling_intern_vit.py
CHANGED
|
@@ -279,6 +279,7 @@ class InternVisionEncoder(nn.Module):
|
|
| 279 |
class InternVisionModel(PreTrainedModel):
|
| 280 |
main_input_name = 'pixel_values'
|
| 281 |
config_class = InternVisionConfig
|
|
|
|
| 282 |
|
| 283 |
def __init__(self, config: InternVisionConfig):
|
| 284 |
super().__init__(config)
|
|
|
|
| 279 |
class InternVisionModel(PreTrainedModel):
|
| 280 |
main_input_name = 'pixel_values'
|
| 281 |
config_class = InternVisionConfig
|
| 282 |
+
_no_split_modules = ['InternAttention']
|
| 283 |
|
| 284 |
def __init__(self, config: InternVisionConfig):
|
| 285 |
super().__init__(config)
|