Upload 8 files

imagerater/config.json

@@ -0,0 +1,35 @@
+{
+    "architectures": [
+        "ViTForImageClassification"
+    ],
+    "attention_probs_dropout_prob": 0,
+    "encoder_stride": 16,
+    "hidden_act": "gelu",
+    "hidden_dropout_prob": 0,
+    "hidden_size": 768,
+    "id2label": {
+        "0": "cute",
+        "1": "erotic",
+        "2": "sexy"
+    },
+    "image_size": 224,
+    "initializer_range": 0.02,
+    "intermediate_size": 3072,
+    "label2id": {
+        "cute": "0",
+        "erotic": "1",
+        "sexy": "2"
+    },
+    "layer_norm_eps": 1e-12,
+    "model_type": "vit",
+    "num_attention_heads": 12,
+    "num_channels": 3,
+    "num_hidden_layers": 12,
+    "patch_size": 16,
+    "pooler_act": "tanh",
+    "pooler_output_size": 768,
+    "problem_type": "single_label_classification",
+    "qkv_bias": true,
+    "torch_dtype": "float32",
+    "transformers_version": "4.54.0"
+}

imagerater/model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:940067457bd70b7b219c50033071384c673843df476402c68bcf7dfb88e1aeec
+size 343227052

imagerater/preprocessor_config.json

@@ -0,0 +1,23 @@
+{
+    "do_convert_rgb": null,
+    "do_normalize": true,
+    "do_rescale": true,
+    "do_resize": true,
+    "image_mean": [
+        0.5,
+        0.5,
+        0.5
+    ],
+    "image_processor_type": "ViTImageProcessor",
+    "image_std": [
+        0.5,
+        0.5,
+        0.5
+    ],
+    "resample": 2,
+    "rescale_factor": 0.00392156862745098,
+    "size": {
+        "height": 224,
+        "width": 224
+    }
+}

segmentator/anime-segmentation.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2c8f6b9a77386c54dcdbf55b6c917108c4bdf4328abca9152c7bce5727b74d18
+size 204275908

sketchextractor/anime2sketch.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ccabdcc3f5cf3c07cf65d58776acb21df7dfda825cdc70c9766a93fd62bfc488
+size 217631959

wdtagger/config.json

@@ -0,0 +1,37 @@
+{
+    "architecture": "swinv2_base_window8_256",
+    "num_classes": 10861,
+    "num_features": 1024,
+    "global_pool": "avg",
+    "model_args": {
+        "act_layer": "gelu_tanh",
+        "img_size": 448,
+        "window_size": 14
+    },
+    "pretrained_cfg": {
+        "custom_load": false,
+        "input_size": [
+            3,
+            448,
+            448
+        ],
+        "fixed_input_size": false,
+        "interpolation": "bicubic",
+        "crop_pct": 1,
+        "crop_mode": "center",
+        "mean": [
+            0.5,
+            0.5,
+            0.5
+        ],
+        "std": [
+            0.5,
+            0.5,
+            0.5
+        ],
+        "num_classes": 10861,
+        "pool_size": null,
+        "first_conv": null,
+        "classifier": null
+    }
+}

wdtagger/model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:39fe04000d5907a0a7616df54af744d43fbd70b5fba7abc3135d3675734c03ef
+size 392149220