Mirrors/ComfyUI_SuperScaler
1
0
Fork 0
mirror of https://github.com/tritant/ComfyUI_SuperScaler.git synced 2025-12-14 15:47:42 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix for Qwen-image-edit

Browse Source
This commit is contained in:
tritant
2025-11-07 07:09:00 +01:00
committed by GitHub
parent 914f62fa36
commit fecf9a9066
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
superscaler.py
View File

@@ -212,7 +212,14 @@ class SuperScaler:
denoise=denoise
)[0]
tile_refined = vae.decode(tile_refined_latent["samples"]) # Sortie en NHWC
tile_refined = vae.decode(tile_refined_latent["samples"]) # Sortie en NHWC, mais avec 5 dimensions
# --- CORRECTION V71: Nettoyage dimensionnel pour les modèles 5D (Qwen) ---
# 1. Squeeze pour retirer les dimensions de taille 1 indésirables (e.g., T=1)
if tile_refined.dim() > 4:
tile_refined = tile_refined.squeeze(1) # Tente de retirer la dimension T
# 2. Permute vers NCHW (pour les calculs suivants) ET move to GPU
tile_refined = tile_refined.permute(0, 3, 1, 2).to(device) # NHWC -> NCHW ET move to GPU
# 5. Appliquer le masque de fondu et ré-assembler