Test

02.forge.sh

@@ -59,7 +59,7 @@ sl_folder ${SD02_DIR}/forge/models GFPGAN ${BASE_DIR}/models gfpgan
 sl_folder ${SD02_DIR}/forge/models LDSR ${BASE_DIR}/models ldsr
 sl_folder ${SD02_DIR}/forge/models ControlNet ${BASE_DIR}/models controlnet
 
-sl_folder ${SD02_DIR}/forge output ${BASE_DIR}/outputs 02-sd-webui
+sl_folder ${SD02_DIR}/forge outputs ${BASE_DIR}/outputs 02-sd-webui
 
 # Run webUI
 echo "Run Stable-Diffusion-WebUI-forge"

02.sh

@@ -46,6 +46,8 @@ pip install onnxruntime-gpu --extra-index-url https://aiinfra.pkgs.visualstudio.
 pip install insightface
 pip install basicsr
 pip install xformers --index-url https://download.pytorch.org/whl/cu121
+pip install albumentations==1.4.3
+pip install pydantic==1.10.15
 
 
 # Merge Models, vae, lora, and hypernetworks, and outputs
@@ -62,7 +64,7 @@ sl_folder ${SD02_DIR}/webui/models GFPGAN ${BASE_DIR}/models gfpgan
 sl_folder ${SD02_DIR}/webui/models LDSR ${BASE_DIR}/models ldsr
 sl_folder ${SD02_DIR}/webui/models ControlNet ${BASE_DIR}/models controlnet
 
-sl_folder ${SD02_DIR}/webui output ${BASE_DIR}/outputs 02-sd-webui
+sl_folder ${SD02_DIR}/webui outputs ${BASE_DIR}/outputs 02-sd-webui
 
 # run webUI
 echo "Run Stable-Diffusion-WebUI"

05.sh

@@ -46,6 +46,8 @@ conda install -c nvidia cuda-cudart --solver=libmamba -y
 pip install onnxruntime-gpu
 pip install insightface torch torchvision
 
+install_requirements ${SD05_DIR}/ComfyUI/custom_nodes
+
 #clean old venv if it still exists
 if [ -d ${SD05_DIR}/venv ]; then
     rm -rf ${SD05_DIR}/venv

Dockerfile

@@ -21,6 +21,7 @@ RUN apt-get update -y -q=2 && \
     cmake \
     build-essential \
     python3-opencv \
+    ffmpeg \
     libopencv-dev \
     dotnet-sdk-8.0 \
     git && \

functions.sh

@@ -52,4 +52,23 @@ sed 's/\// /g') | cut -f1) ]; then
     git reset --hard HEAD
     git pull -X ours
 fi
+}
+
+# Fonction récursive pour installer les requirements.txt
+install_requirements() {
+    local directory="$1"
+    local requirements_file="$directory/requirements.txt"
+
+    if [ -f "$requirements_file" ]; then
+        echo "Installation des dépendances dans $directory ..."
+        pip install -r "$requirements_file"
+        echo "Dépendances installées avec succès dans $directory."
+    fi
+
+    # Parcours récursif des sous-dossiers
+    for subdir in "$directory"/*; do
+        if [ -d "$subdir" ]; then
+            install_requirements "$subdir"
+        fi
+    done
 }