initial commit

Signed-off-by: Robert Günzler <r@gnzler.io>

2 files changed, 56 insertions, 0 deletions

diff --git a/bundles/vast-base/items.py b/bundles/vast-base/items.py
new file mode 100644
index 0000000..5d705a9
--- /dev/null
+++ b/bundles/vast-base/items.py
@@ -0,0 +1,32 @@
+from shlex import quote
+
+if node.os != "debian":
+    raise BundleError(f"{node.name}: OS {node.os} is not supported.")
+
+
+files = {
+    "/workspace/storage/stable_diffusion/models/ckpt/v1-5-pruned-emaonly.ckpt": {
+        "delete": True,
+    },
+
+}
+
+# ranbooru_dir = "/workspace/stable-diffusion-webui/extensions/sd-webui-ranbooru"
+# ranbooru_url = "https://github.com/Inzaniak/sd-webui-ranbooru/archive/master.tar.gz"
+#
+# actions = {
+#     "download_ranbooru": {
+#         "command": f"aria2c -o /workspace/master.tgz {quote(ranbooru_url)} && tar xf /workspace/master.tgz --strip-components=1 {quote(ranbooru_dir)}; rm -rf /workspace/master.tgz;",
+#         "needs": {"pkg_apt:aria2"},
+#         "unless": f"test -d {ranbooru_dir}/README.md",
+#     },
+# }
+#
+modeldir = node.metadata.get("vast/model-dir", "/tmp/")
+for url, opts in node.metadata.get("vast/models", {}).items():
+    print('???', opts)
+    actions["download_" + opts["filename"]] = {
+        "command": f"aria2c -d {quote(modeldir)} {quote(url)}{f"?token={repo.vault.cmd("rbw get civitai/apikey")}" if "civitai.com" in url else ""}",
+        "needs": {"pkg_apt:aria2"},
+        "unless": f"printf '{opts["sha256"]}  {modeldir}/{opts["filename"]}' | sha256sum -c -",
+    }
diff --git a/bundles/vast-base/metadata.py b/bundles/vast-base/metadata.py
new file mode 100644
index 0000000..23196f3
--- /dev/null
+++ b/bundles/vast-base/metadata.py
@@ -0,0 +1,24 @@
+defaults = {
+    "apt": {
+        "packages": {
+            "aria2": {},
+        },
+    },
+    "vast": {
+        "model-dir": "/workspace/storage/stable_diffusion/models/ckpt/",
+        "models": {
+            "https://civitai.com/api/download/models/324619": {
+                "filename": "autismmixSDXL_autismmixPony.safetensors",
+                "sha256": "821aa5537f8ddafdbf963827551865c31c5bbfab1abe7925cb5f006c8f71e485",
+            },
+            "https://civitai.com/api/download/models/534642": {
+                "filename": "ponyRealism_v21MainVAE.safetensors",
+                "sha256": "059934FF58033C5ABB606B6B98F969FFDF7E7312D352C3BE47A0440BA28D303B",
+            },
+            "https://civitai.com/api/download/models/559994": {
+                "filename": "HassakuXL.safetensors",
+                "sha256": "4F302F0C92213BE34BA6404E1D0D2140922C6A2FA26C5F4FFACD118207D7D927",
+            },
+        },
+    },
+}