tensorgirl
/

TFaugvit

Image Classification

Inference Endpoints

Model card Files Files and versions Community

tensorgirl commited on Sep 21, 2023

Commit

1dd077d

·

1 Parent(s): c452099

Update config.json

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -4,8 +4,8 @@
     "gViTForImageClassification"
   ],
   "auto_map": {
-    "AutoConfig": "tensorgirl/TFaugvit--augvit_config.AugViTConfig",
-    "TFAutoModelForImageClassification": "tensorgirl/TFaugvit--augvit_model.AugViTForImageClassification"
   },
   "depth": 2,
   "dim": 128,

     "gViTForImageClassification"
   ],
   "auto_map": {
+    "AutoConfig": "augvit_config.AugViTConfig",
+    "TFAutoModelForImageClassification": "augvit_model.AugViTForImageClassification"
   },
   "depth": 2,
   "dim": 128,