kd13
/

vit-nano-patch16-224

Image Classification

custom_vit_nano

Model card Files Files and versions

kd13 commited on 4 days ago

Commit

c890784

·

verified ·

1 Parent(s): 4a59940

Update modeling_vit.py

Files changed (1) hide show

modeling_vit.py +1 -1

modeling_vit.py CHANGED Viewed

@@ -134,7 +134,7 @@ class Block(nn.Module):
         return x
 class CustomViTNanoPreTrainedModel(PreTrainedModel):
-    config_class = CustomViTNanoV2Config
     base_model_prefix = "custom_vit"
     main_input_name = "pixel_values"
     _no_split_modules = ["Block"]

         return x
 class CustomViTNanoPreTrainedModel(PreTrainedModel):
+    config_class = CustomViTNanoConfig
     base_model_prefix = "custom_vit"
     main_input_name = "pixel_values"
     _no_split_modules = ["Block"]