version: v0.13.0

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "wdtagger"
-version = "0.12.0"
+version = "0.13.0"
 description = "A simple and easy-to-use wrapper for the tagger model created by [SmilingWolf](https://github.com/SmilingWolf) which is specifically designed for tagging anime illustrations."
 authors = [{ name = "Jianqi Pan", email = "jannchie@gmail.com" }]
 readme = "README.md"
@@ -39,18 +39,29 @@ ignore = [
 [tool.pyright]
 
 [tool.uv]
-conflicts = [[{ extra = "cpu" }, { extra = "gpu" }]]
+conflicts = [
+    [
+        { extra = "cpu" },
+        { extra = "gpu" },
+        { extra = "cuda11" },
+        { extra = "cuda12" },
+    ],
+]
 package = true
 
 [tool.uv.sources]
 torch = [
     { index = "torch-cpu", extra = "cpu" },
     { index = "torch-gpu", extra = "gpu" },
+    { index = "torch-cuda11", extra = "cuda11" },
+    { index = "torch-cuda12", extra = "cuda12" },
 ]
 
 torchvision = [
     { index = "torch-cpu", extra = "cpu" },
     { index = "torch-gpu", extra = "gpu" },
+    { index = "torch-cuda11", extra = "cuda11" },
+    { index = "torch-cuda12", extra = "cuda12" },
 ]
 
 [[tool.uv.index]]
@@ -63,6 +74,16 @@ name = "torch-gpu"
 url = "https://download.pytorch.org/whl/cu124"
 explicit = true
 
+[[tool.uv.index]]
+name = "torch-cuda11"
+url = "https://download.pytorch.org/whl/cu118"
+explicit = true
+
+[[tool.uv.index]]
+name = "torch-cuda12"
+url = "https://download.pytorch.org/whl/cu124"
+explicit = true
+
 [build-system]
 requires = ["hatchling"]
 build-backend = "hatchling.build"