file

YanWenKun · Dec 7, 2024 · 0fc6a1d · 0fc6a1d
1 parent 2776078
commit 0fc6a1d
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/pt24-cu121/stage2.sh b/pt24-cu121/stage2.sh
@@ -39,8 +39,7 @@ git reset --hard "1.9"
 cd "$workdir"
 mv  python_embeded  Comfy3D_WinPortable/python_embeded
 
-# Copy & Replace start script files
-
+# Copy start script files
 cp -rf "$workdir"/attachments/* \
     "$workdir"/Comfy3D_WinPortable/
 
@@ -56,6 +55,11 @@ $gcs https://github.com/ltdrdata/ComfyUI-Impact-Subpack.git impact_subpack
 cd "$workdir"/Comfy3D_WinPortable
 ./python_embeded/python.exe -s -B ComfyUI/main.py --quick-test-for-ci --cpu
 
+# Copy u2net model files needed by rembg (to avoid download at first start)
+cd "$workdir"/Comfy3D_WinPortable
+mkdir extras
+cp ~/.u2net/u2net.onnx ./extras/u2net.onnx
+
 # Copy/Move example files of 3D-Pack
 mkdir -p "$workdir"/Comfy3D_WinPortable/ComfyUI/user/default/workflows