cs-giung
/

vit-base-patch16-imagenet21k-augreg

Image Classification

Inference Endpoints

Model card Files Files and versions Community

cs-giung commited on Jun 1

Commit

a839137

•

1 Parent(s): b86df05

Update config.json

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -1,7 +1,7 @@
 {
   "_name_or_path": "cs-giung/vit-base-patch16-imagenet21k-augreg",
   "architectures": [
-    "ViTModel"
   ],
   "attention_probs_dropout_prob": 0.0,
   "hidden_act": "gelu",

 {
   "_name_or_path": "cs-giung/vit-base-patch16-imagenet21k-augreg",
   "architectures": [
+    "ViTForImageClassification"
   ],
   "attention_probs_dropout_prob": 0.0,
   "hidden_act": "gelu",