Fix missing dependency for docker image and add flux dev as resource

Acly · Sep 14, 2024 · 8f7395c · 8f7395c
1 parent 13e613f
commit 8f7395c
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 2 deletions.
diff --git a/ai_diffusion/__init__.py b/ai_diffusion/__init__.py
@@ -1,6 +1,6 @@
 """Generative AI plugin for Krita using Stable Diffusion"""
 
-__version__ = "1.24.0"
+__version__ = "1.24.1"
 
 import importlib.util
 

diff --git a/ai_diffusion/resources.py b/ai_diffusion/resources.py
@@ -481,6 +481,15 @@ def __hash__(self):
             ): "https://huggingface.co/misri/zavychromaxl_v80/resolve/main/zavychromaxl_v80.safetensors"
         },
     ),
+    ModelResource(
+        "Flux [dev]",
+        ResourceId(ResourceKind.checkpoint, Arch.flux, "flux-dev"),
+        {
+            Path(
+                "models/checkpoints/flux1-dev-fp8.safetensors"
+            ): "https://huggingface.co/Comfy-Org/flux1-dev/resolve/main/flux1-dev-fp8.safetensors"
+        },
+    ),
     ModelResource(
         "Flux [schnell]",
         ResourceId(ResourceKind.checkpoint, Arch.flux, "flux-schnell"),

diff --git a/scripts/docker/Dockerfile b/scripts/docker/Dockerfile
@@ -41,6 +41,7 @@ COPY ./ComfyUI /ComfyUI/
 RUN python3 -m pip install -r /ComfyUI/requirements.txt && \
     python3 -m pip install -r /ComfyUI/custom_nodes/comfyui_controlnet_aux/requirements.txt && \
     python3 -m pip install -r /ComfyUI/custom_nodes/comfyui-tooling-nodes/requirements.txt && \
+    python3 -m pip install -r /ComfyUI/custom_nodex/ComfyUI-GGUF/requirements.txt && \
     python3 -m pip install insightface && \
     python3 -m pip cache purge
 
@@ -55,7 +56,8 @@ RUN ln -s /workspace/models /models
 FROM base
 
 # Install additional dependencies (convenience for manual interaction)
-RUN apt install -y --no-install-recommends \
+RUN apt update && \
+    apt install -y --no-install-recommends \
     nginx \
     net-tools \
     inetutils-ping \