wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-03-16 22:58:19 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Merge bfdfc7d11c into 1c5db7397d

Browse Source
This commit is contained in:
Pam 2026-03-14 18:36:45 -04:00 committed by GitHub
commit 367cf779ef
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
comfy_extras/nodes_eps.py
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)