monurcan
/

monurcan commited on
Commit
c8e50c3
·
verified ·
1 Parent(s): b3873cf

Update modeling_aimv2.py

Browse files
Files changed (1) hide show
  1. modeling_aimv2.py +1 -1
modeling_aimv2.py CHANGED
@@ -339,7 +339,7 @@ class AIMv2Transformer(nn.Module):
339
 
340
  class AIMv2PretrainedModel(PreTrainedModel):
341
  config_class = AIMv2Config
342
- base_model_prefix = "aimv2"
343
  supports_gradient_checkpointing = True
344
  main_input_name = "pixel_values"
345
  _no_split_modules = ["AIMv2ViTPreprocessor", "AIMv2Block"]
 
339
 
340
  class AIMv2PretrainedModel(PreTrainedModel):
341
  config_class = AIMv2Config
342
+ base_model_prefix = "aimv2onur"
343
  supports_gradient_checkpointing = True
344
  main_input_name = "pixel_values"
345
  _no_split_modules = ["AIMv2ViTPreprocessor", "AIMv2Block"]