Update docker image with upscaling requirements

Acly · Oct 28, 2023 · c81184b · c81184b
1 parent a12dc8c
commit c81184b
Show file tree

Hide file tree

Showing 3 changed files with 6 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.4.0"
+__version__ = "1.5.0"
 
 from . import util
 from .settings import Settings, Setting, settings, PerformancePreset, ServerBackend, ServerMode

diff --git a/scripts/docker.py b/scripts/docker.py
@@ -19,7 +19,10 @@
 
 def all_models():
     return chain(
-        resources.required_models, resources.optional_models, resources.default_checkpoints
+        resources.required_models,
+        resources.optional_models,
+        resources.default_checkpoints,
+        resources.upscale_models,
     )
 
 

diff --git a/scripts/docker/Dockerfile b/scripts/docker/Dockerfile
@@ -66,6 +66,7 @@ RUN git clone https://github.com/comfyanonymous/ComfyUI.git /ComfyUI && \
 RUN git clone https://github.com/ltdrdata/ComfyUI-Manager.git /ComfyUI/custom_nodes/ComfyUI-Manager && \
     git clone https://github.com/Fannovel16/comfyui_controlnet_aux /ComfyUI/custom_nodes/comfyui_controlnet_aux && \
     git clone https://github.com/cubiq/ComfyUI_IPAdapter_plus /ComfyUI/custom_nodes/ComfyUI_IPAdapter_plus && \
+    git clone https://github.com/ssitu/ComfyUI_UltimateSDUpscale /ComfyUI/custom_nodes/ComfyUI_UltimateSDUpscale --recursive && \
     git clone https://github.com/Acly/comfyui-tooling-nodes /ComfyUI/custom_nodes/comfyui-tooling-nodes && \
     pip3 install -r /ComfyUI/custom_nodes/ComfyUI-Manager/requirements.txt && \
     pip3 install -r /ComfyUI/custom_nodes/comfyui_controlnet_aux/requirements.txt && \