From bfdfc7d11c3d25d75804f8b010d2b0c0d971ad5d Mon Sep 17 00:00:00 2001 From: Pam Date: Wed, 11 Mar 2026 05:43:44 +0500 Subject: [PATCH] Fix for TemporalScoreRescaling node when batch_size > 1 --- comfy_extras/nodes_eps.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/comfy_extras/nodes_eps.py b/comfy_extras/nodes_eps.py index 0fb3871c8..07a4349a7 100644 --- a/comfy_extras/nodes_eps.py +++ b/comfy_extras/nodes_eps.py @@ -133,11 +133,11 @@ class TemporalScoreRescaling(io.ComfyNode): def temporal_score_rescaling(args): denoised = args["denoised"] x = args["input"] - sigma = args["sigma"] + sigma = args["sigma"][0] curr_model = args["model"] # No rescaling (r = 1) or no noise - if tsr_k == 1 or sigma == 0: + if tsr_k == 1 or sigma.item() == 0: return denoised model_sampling = curr_model.current_patcher.get_model_object("model_sampling") @@ -145,7 +145,7 @@ class TemporalScoreRescaling(io.ComfyNode): snr = (2 * half_log_snr).exp() # No rescaling needed (r = 1) - if snr == 0: + if snr.item() == 0: return denoised rescaling_r = compute_tsr_rescaling_factor(snr, tsr_k, tsr_variance)