apple
/

aimv2-huge-patch14-448

Image Feature Extraction

aimv2_vision_model

Model card Files Files and versions

michalk8 commited on Nov 28, 2024

Commit

6299ef9

·

1 Parent(s): 30974f9

Add `_no_split_modules`

Files changed (1) hide show

modeling_aimv2.py +1 -0

modeling_aimv2.py CHANGED Viewed

@@ -149,6 +149,7 @@ class AIMv2PretrainedModel(PreTrainedModel):
     config_class = AIMv2Config
     base_model_prefix = "aimv2"
     main_input_name = "pixel_values"
     _supports_sdpa = True

     config_class = AIMv2Config
     base_model_prefix = "aimv2"
     main_input_name = "pixel_values"
+    _no_split_modules = ["AIMv2ViTPreprocessor", "AIMv2Block"]
     _supports_sdpa = True