wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-17 00:43:48 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Update nodes_qwen.py

Browse Source 
Fixed a bug in TextEncodeQwenImageEditPlus where the rescaling would not preserve the original image ratio.
This commit is contained in:
dchatel 2025-10-04 08:22:11 +02:00 committed by GitHub
parent bbd683098e
commit 7d559ba23e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
comfy_extras/nodes_qwen.py
View File

@ -87,10 +87,15 @@ class TextEncodeQwenImageEditPlus(io.ComfyNode):
s = comfy.utils.common_upscale(samples, width, height, "area", "disabled") s = comfy.utils.common_upscale(samples, width, height, "area", "disabled")
images_vl.append(s.movedim(1, -1)) images_vl.append(s.movedim(1, -1))
if vae is not None: if vae is not None:
# compute optimal scale factor to get as close to 1024x1024 as possible while keeping dimensions multiples of 8
total = int(1024 * 1024) total = int(1024 * 1024)
scale_by = math.sqrt(total / (samples.shape[3] * samples.shape[2])) s_ideal = math.sqrt(total / (samples.shape[3] * samples.shape[2]))
width = round(samples.shape[3] * scale_by / 8.0) * 8 common_divisor = math.gcd(samples.shape[3], samples.shape[2])
height = round(samples.shape[2] * scale_by / 8.0) * 8 p = samples.shape[3] // common_divisor
lcm = abs(p * 8) // math.gcd(p, 8)
s_optimal = (round(s_ideal * samples.shape[3] / lcm) * lcm) / samples.shape[3]
width = int(s_optimal * samples.shape[3])
height = int(s_optimal * samples.shape[2])
s = comfy.utils.common_upscale(samples, width, height, "area", "disabled") s = comfy.utils.common_upscale(samples, width, height, "area", "disabled")
ref_latents.append(vae.encode(s.movedim(1, -1)[:, :, :, :3])) ref_latents.append(vae.encode(s.movedim(1, -1)[:, :, :, :3]))