From 38be6c13f76e893cf47636257071b440dec0c5b2 Mon Sep 17 00:00:00 2001 From: Robert Günzler Date: Sun, 7 Sep 2025 17:32:13 +0200 Subject: initial commit MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Robert Günzler --- bundles/vast-base/items.py | 32 ++++++++++++++++++++++++++++++++ 1 file changed, 32 insertions(+) create mode 100644 bundles/vast-base/items.py (limited to 'bundles/vast-base/items.py') 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 -", + } -- cgit 1.4.1