kd13
/

vit-nano-patch16-224

Image Classification

custom_vit_nano

Model card Files Files and versions

kd13 commited on 5 days ago

Commit

ba314eb

·

verified ·

1 Parent(s): 8d90afc

Update configuration_vit.py

Files changed (1) hide show

configuration_vit.py +2 -2

configuration_vit.py CHANGED Viewed

@@ -1,6 +1,6 @@
 from transformers import PretrainedConfig
-class CustomViTNanoV2Config(PretrainedConfig):
     model_type = "custom_vit_nano"
     def __init__(
@@ -29,4 +29,4 @@ class CustomViTNanoV2Config(PretrainedConfig):
         self.stem_channels = stem_channels
         self.dropout = dropout
-CustomViTNanoV2Config.register_for_auto_class()

 from transformers import PretrainedConfig
+class CustomViTNanoConfig(PretrainedConfig):
     model_type = "custom_vit_nano"
     def __init__(
         self.stem_channels = stem_channels
         self.dropout = dropout
+CustomViTNanoConfig.register_for_auto_class()