From 411ad961b081a7824b4320a563ef0f6e1d4c70d8 Mon Sep 17 00:00:00 2001 From: silveroxides Date: Wed, 15 Jul 2026 21:06:17 +0200 Subject: [PATCH] perf: preserve dtype during visual resize --- comfy_extras/nodes_cond.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/comfy_extras/nodes_cond.py b/comfy_extras/nodes_cond.py index 0a2dc5ea2..262e5bd30 100644 --- a/comfy_extras/nodes_cond.py +++ b/comfy_extras/nodes_cond.py @@ -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):