p1atdev
/

wd-swinv2-tagger-v3-hf

Image Classification

Model card Files Files and versions Community

p1atdev commited on Mar 8, 2024

Commit

be81ef4

·

verified ·

1 Parent(s): bfb6030

Upload config.json

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -4,7 +4,7 @@
   ],
   "attention_probs_dropout_prob": 0.0,
   "auto_map": {
-    "AutoImageProcessor": "image_processing_tagger.WDv14TaggerImageProcessor"
   },
   "depths": [
     2,
@@ -21779,7 +21779,7 @@
     "stage3",
     "stage4"
   ],
-  "torch_dtype": "bfloat16",
   "transformers_version": "4.38.1",
   "use_absolute_embeddings": false,
   "window_size": 14

   ],
   "attention_probs_dropout_prob": 0.0,
   "auto_map": {
+    "AutoImageProcessor": "image_processing_tagger.WDTaggerImageProcessor"
   },
   "depths": [
     2,
     "stage3",
     "stage4"
   ],
+  "torch_dtype": "float32",
   "transformers_version": "4.38.1",
   "use_absolute_embeddings": false,
   "window_size": 14