kd13
/

vit-nano-patch16-224

Image Classification

custom_vit_nano

Model card Files Files and versions

kd13 commited on 4 days ago

Commit

d0ce4da

·

verified ·

1 Parent(s): 205ea67

Update modeling_vit.py

Files changed (1) hide show

modeling_vit.py +1 -1

modeling_vit.py CHANGED Viewed

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

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