czczup commited on
Commit
c3ac410
1 Parent(s): 65437ee

Update modeling_intern_vit.py

Browse files
Files changed (1) hide show
  1. modeling_intern_vit.py +1 -1
modeling_intern_vit.py CHANGED
@@ -279,7 +279,7 @@ class InternVisionEncoder(nn.Module):
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)
 
279
  class InternVisionModel(PreTrainedModel):
280
  main_input_name = 'pixel_values'
281
  config_class = InternVisionConfig
282
+ _no_split_modules = ['InternVisionEncoderLayer']
283
 
284
  def __init__(self, config: InternVisionConfig):
285
  super().__init__(config)