From 6ff516dbe97f579f6a2af465309064e183c8dc43 Mon Sep 17 00:00:00 2001
From: pixelass <greg@pixelass.com>
Date: Mon, 12 Feb 2024 09:52:43 +0100
Subject: [PATCH] feat: add wd14 caption models

---
 .../wd-v1-4-convnext-tagger-v2/example.txt       |  1 +
 .../wd-v1-4-convnext-tagger-v2/info.json5        | 16 ++++++++++++++++
 .../wd-v1-4-convnextv2-tagger-v2/example.txt     |  1 +
 .../wd-v1-4-convnextv2-tagger-v2/info.json5      | 16 ++++++++++++++++
 .../wd-v1-4-moat-tagger-v2/example.txt           |  1 +
 .../wd-v1-4-moat-tagger-v2/info.json5            | 16 ++++++++++++++++
 .../wd-v1-4-swinv2-tagger-v2/example.txt         |  1 +
 .../wd-v1-4-swinv2-tagger-v2/info.json5          | 16 ++++++++++++++++
 8 files changed, 68 insertions(+)
 create mode 100644 files/caption/wd14/SmilingWolf/wd-v1-4-convnext-tagger-v2/example.txt
 create mode 100644 files/caption/wd14/SmilingWolf/wd-v1-4-convnext-tagger-v2/info.json5
 create mode 100644 files/caption/wd14/SmilingWolf/wd-v1-4-convnextv2-tagger-v2/example.txt
 create mode 100644 files/caption/wd14/SmilingWolf/wd-v1-4-convnextv2-tagger-v2/info.json5
 create mode 100644 files/caption/wd14/SmilingWolf/wd-v1-4-moat-tagger-v2/example.txt
 create mode 100644 files/caption/wd14/SmilingWolf/wd-v1-4-moat-tagger-v2/info.json5
 create mode 100644 files/caption/wd14/SmilingWolf/wd-v1-4-swinv2-tagger-v2/example.txt
 create mode 100644 files/caption/wd14/SmilingWolf/wd-v1-4-swinv2-tagger-v2/info.json5

diff --git a/files/caption/wd14/SmilingWolf/wd-v1-4-convnext-tagger-v2/example.txt b/files/caption/wd14/SmilingWolf/wd-v1-4-convnext-tagger-v2/example.txt
new file mode 100644
index 0000000..d00589b
--- /dev/null
+++ b/files/caption/wd14/SmilingWolf/wd-v1-4-convnext-tagger-v2/example.txt
@@ -0,0 +1 @@
+monochrome, outdoors, signature, tree, no humans, scenery, road, bush
\ No newline at end of file
diff --git a/files/caption/wd14/SmilingWolf/wd-v1-4-convnext-tagger-v2/info.json5 b/files/caption/wd14/SmilingWolf/wd-v1-4-convnext-tagger-v2/info.json5
new file mode 100644
index 0000000..2c51627
--- /dev/null
+++ b/files/caption/wd14/SmilingWolf/wd-v1-4-convnext-tagger-v2/info.json5
@@ -0,0 +1,16 @@
+{
+  type: "trained",
+  architecture: "ONNX",
+  title: "WD 1.4 ConvNext Tagger V2",
+  author: "Smiling Wolf",
+  link: "https://huggingface.co/SmilingWolf/wd-v1-4-convnext-tagger-v2",
+  license: "apache-2.0",
+  files: [
+    {
+      filename: "model.onnx",
+    },
+    {
+      filename: "selected_tags.csv",
+    }
+  ],
+}
diff --git a/files/caption/wd14/SmilingWolf/wd-v1-4-convnextv2-tagger-v2/example.txt b/files/caption/wd14/SmilingWolf/wd-v1-4-convnextv2-tagger-v2/example.txt
new file mode 100644
index 0000000..d00589b
--- /dev/null
+++ b/files/caption/wd14/SmilingWolf/wd-v1-4-convnextv2-tagger-v2/example.txt
@@ -0,0 +1 @@
+monochrome, outdoors, signature, tree, no humans, scenery, road, bush
\ No newline at end of file
diff --git a/files/caption/wd14/SmilingWolf/wd-v1-4-convnextv2-tagger-v2/info.json5 b/files/caption/wd14/SmilingWolf/wd-v1-4-convnextv2-tagger-v2/info.json5
new file mode 100644
index 0000000..6deac55
--- /dev/null
+++ b/files/caption/wd14/SmilingWolf/wd-v1-4-convnextv2-tagger-v2/info.json5
@@ -0,0 +1,16 @@
+{
+  type: "trained",
+  architecture: "ONNX",
+  title: "WD 1.4 ConvNextV2 Tagger V2",
+  author: "Smiling Wolf",
+  link: "https://huggingface.co/SmilingWolf/wd-v1-4-convnextv2-tagger-v2",
+  license: "apache-2.0",
+  files: [
+    {
+      filename: "model.onnx",
+    },
+    {
+      filename: "selected_tags.csv",
+    }
+  ],
+}
diff --git a/files/caption/wd14/SmilingWolf/wd-v1-4-moat-tagger-v2/example.txt b/files/caption/wd14/SmilingWolf/wd-v1-4-moat-tagger-v2/example.txt
new file mode 100644
index 0000000..d00589b
--- /dev/null
+++ b/files/caption/wd14/SmilingWolf/wd-v1-4-moat-tagger-v2/example.txt
@@ -0,0 +1 @@
+monochrome, outdoors, signature, tree, no humans, scenery, road, bush
\ No newline at end of file
diff --git a/files/caption/wd14/SmilingWolf/wd-v1-4-moat-tagger-v2/info.json5 b/files/caption/wd14/SmilingWolf/wd-v1-4-moat-tagger-v2/info.json5
new file mode 100644
index 0000000..dc3c546
--- /dev/null
+++ b/files/caption/wd14/SmilingWolf/wd-v1-4-moat-tagger-v2/info.json5
@@ -0,0 +1,16 @@
+{
+  type: "trained",
+  architecture: "ONNX",
+  title: "WD 1.4 MOAT Tagger V2",
+  author: "Smiling Wolf",
+  link: "https://huggingface.co/SmilingWolf/wd-v1-4-moat-tagger-v2",
+  license: "apache-2.0",
+  files: [
+    {
+      filename: "model.onnx",
+    },
+    {
+      filename: "selected_tags.csv",
+    }
+  ],
+}
diff --git a/files/caption/wd14/SmilingWolf/wd-v1-4-swinv2-tagger-v2/example.txt b/files/caption/wd14/SmilingWolf/wd-v1-4-swinv2-tagger-v2/example.txt
new file mode 100644
index 0000000..d00589b
--- /dev/null
+++ b/files/caption/wd14/SmilingWolf/wd-v1-4-swinv2-tagger-v2/example.txt
@@ -0,0 +1 @@
+monochrome, outdoors, signature, tree, no humans, scenery, road, bush
\ No newline at end of file
diff --git a/files/caption/wd14/SmilingWolf/wd-v1-4-swinv2-tagger-v2/info.json5 b/files/caption/wd14/SmilingWolf/wd-v1-4-swinv2-tagger-v2/info.json5
new file mode 100644
index 0000000..655e6af
--- /dev/null
+++ b/files/caption/wd14/SmilingWolf/wd-v1-4-swinv2-tagger-v2/info.json5
@@ -0,0 +1,16 @@
+{
+  type: "trained",
+  architecture: "ONNX",
+  title: "WD 1.4 SwinV2 Tagger V2",
+  author: "Smiling Wolf",
+  link: "https://huggingface.co/SmilingWolf/wd-v1-4-swinv2-tagger-v2",
+  license: "apache-2.0",
+  files: [
+    {
+      filename: "model.onnx",
+    },
+    {
+      filename: "selected_tags.csv",
+    }
+  ],
+}