From 8076147646820a701a7d50e311dfc221c6e0473a Mon Sep 17 00:00:00 2001
From: YAN Wenkun <code@yanwk.fun>
Date: Tue, 29 Oct 2024 01:55:23 +0800
Subject: [PATCH] bugfix: copy reinstall-deps-for-3d.bat

---
 stage2.sh | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/stage2.sh b/stage2.sh
index 314b9c6..2e4e13c 100644
--- a/stage2.sh
+++ b/stage2.sh
@@ -89,9 +89,13 @@ cp -r ComfyUI/.ci/update_windows/* ./update/
 cp -r ComfyUI/.ci/windows_base_files/* ./
 
 # Replace start script file
-rm "$workdir"/ComfyUI_Windows_portable/run_nvidia_gpu.bat
-cp "$workdir"/attachments/run_nvidia_gpu.bat \
-    "$workdir"/ComfyUI_Windows_portable/run_nvidia_gpu.bat
+# Check if ComfyUI break-changes
+if [ ! -f "$workdir"/ComfyUI_Windows_portable/run_nvidia_gpu.bat ] ; then
+    return 1
+fi ;
+# Copy files under 'attachments' folder
+cp -rf "$workdir"/attachments/* \
+    "$workdir"/ComfyUI_Windows_portable/
 
 du -hd1 "$workdir"