wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-07-18 12:28:17 +08:00
Code Issues Actions 30 Packages Projects Releases Wiki Activity

perf: preserve dtype during visual resize

Browse Source
This commit is contained in:
silveroxides 2026-07-15 21:06:17 +02:00
parent 20a8d3b9ae
commit 411ad961b0
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy_extras/nodes_cond.py
View File

@ -54,13 +54,12 @@ def _resize_visual_tokens(visual, source_grid, target_grid):
if source_grid == target_grid:
return visual
dtype = visual.dtype
batch, _, dimensions = visual.shape
height, width = source_grid
target_height, target_width = target_grid
visual = visual.reshape(batch, height, width, dimensions).permute(0, 3, 1, 2).float()
visual = visual.reshape(batch, height, width, dimensions).permute(0, 3, 1, 2)
visual = F.interpolate(visual, size=(target_height, target_width), mode="bilinear", align_corners=False)
return visual.permute(0, 2, 3, 1).reshape(batch, target_height * target_width, dimensions).to(dtype=dtype)
return visual.permute(0, 2, 3, 1).reshape(batch, target_height * target_width, dimensions)
def _fuse_conditionings(conditionings, tokens, visual_grids, method, block_size, dither_ratio, seed=0):