R1000 commited on
Commit
2738ec7
·
verified ·
1 Parent(s): 65fbafb

Update start.sh

Browse files
Files changed (1) hide show
  1. start.sh +30 -0
start.sh CHANGED
@@ -200,6 +200,36 @@ else
200
  echo "[INFO] KREA2_DOWNLOAD_MODELS=0; model download skipped."
201
  fi
202
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
203
  # ------------------------------------------------------------
204
  # CPU optimization
205
  # ------------------------------------------------------------
 
200
  echo "[INFO] KREA2_DOWNLOAD_MODELS=0; model download skipped."
201
  fi
202
 
203
+ # ------------------------------------------------------------
204
+ # [ADDED] Fix Qwen-Image-2.1-Turbo GGUF for ComfyUI
205
+ # ------------------------------------------------------------
206
+ QWEN_FIX_TARGET="$COMFY/models/unet/Qwen-Image-2.1-Turbo-Q8_0-comfyfix.gguf"
207
+
208
+ # ตรวจสอบว่าโมเดลถูกโหลดลงมาแล้ว และสคริปต์แก้ไขตัวนี้ยังไม่เคยรันมาก่อน
209
+ if [ -f "$QWEN_FIX_TARGET" ]; then
210
+ echo "[INFO] Checking and preparing Qwen-Image GGUF Fix..."
211
+
212
+ # ติดตั้ง dependencies ที่จำเป็นสำหรับสคริปต์ (ถ้าตัว Container ยังไม่มี)
213
+ pip install --no-cache-dir gguf numpy
214
+
215
+ # ดาวน์โหลดสคริปต์แก้บั๊กโดยตรงจาก Hugging Face Repo ของ skyzer4ever
216
+ echo "[DOWNLOAD] Fetching fix_atomicchat_gguf.py..."
217
+ curl -s -L -o /tmp/fix_atomicchat_gguf.py \
218
+ "https://huggingface.co/skyzer4ever/ComfyFix-GGUF-for-Qwen-Image-2.1-Turbo/resolve/main/fix_atomicchat_gguf.py"
219
+
220
+ # สั่งรันสคริปต์แก้โครงสร้างไฟล์ (จะใช้เวลาประมาณ 6 วินาทีในการแก้ไฟล์)
221
+ echo "[FIXING] Running tensor conversion on Qwen GGUF..."
222
+ python /tmp/fix_atomicchat_gguf.py "$QWEN_FIX_TARGET" "$QWEN_FIX_TARGET.fixed"
223
+
224
+ # สลับไฟล์ที่แก้แล้วมาเป็นไฟล์หลักแทน
225
+ if [ -f "$QWEN_FIX_TARGET.fixed" ]; then
226
+ mv "$QWEN_FIX_TARGET.fixed" "$QWEN_FIX_TARGET"
227
+ echo "[OK] Qwen-Image-2.1-Turbo GGUF is now fully patched for ComfyUI!"
228
+ else
229
+ echo "[WARNING] Fix script failed to output a fixed file. Keeping original."
230
+ fi
231
+ fi
232
+
233
  # ------------------------------------------------------------
234
  # CPU optimization
235
  # ------------------------------------------------------------