wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-07-21 23:41:28 +08:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

Merge bfdfc7d11c into 6665515349

Browse Source
This commit is contained in:
Pam
2026-07-20 13:41:43 +08:00
committed by GitHub
parent 6665515349 bfdfc7d11c
commit 20a693a1a1
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
comfy_extras/nodes_eps.py
+3 -3
View File
@@ -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)