Add
Browse files- modeling_aimv2.py +1 -0
modeling_aimv2.py
CHANGED
|
@@ -178,6 +178,7 @@ class AIMv2PretrainedModel(PreTrainedModel):
|
|
| 178 |
config_class = AIMv2Config
|
| 179 |
base_model_prefix = "aimv2"
|
| 180 |
main_input_name = "pixel_values"
|
|
|
|
| 181 |
_supports_sdpa = True
|
| 182 |
|
| 183 |
|
|
|
|
| 178 |
config_class = AIMv2Config
|
| 179 |
base_model_prefix = "aimv2"
|
| 180 |
main_input_name = "pixel_values"
|
| 181 |
+
_no_split_modules = ["AIMv2ViTPreprocessor", "AIMv2Block"]
|
| 182 |
_supports_sdpa = True
|
| 183 |
|
| 184 |
|