xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2026-01-26 06:24:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update nodes.py

Browse Source
This commit is contained in:
kijai 2025-11-23 02:18:47 +02:00
parent 8660778ea1
commit 4dfb85dcc5
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
nodes/nodes.py
View File

@ -2671,7 +2671,10 @@ class TTM_ApplyModel_Wrapper:
if current_step_index != 0 and current_step_index < self.steps:
noisy_latent = self.scale_latent_inpaint(x=x, sigma=torch.tensor([next_sigma]), noise=self.noise.to(x), latent_image=self.reference_samples.to(x))
x = x * (1-self.motion_mask).to(x) + noisy_latent * self.motion_mask.to(x)
if self.motion_mask is not None:
x = x * (1-self.motion_mask).to(x) + noisy_latent * self.motion_mask.to(x)
else:
x = noisy_latent
return executor(x, t, c_concat, c_crossattn, control, transformer_options, **kwargs)