xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2026-04-15 22:56:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

small tweaks

Browse Source
This commit is contained in:
kijai 2025-03-28 17:24:44 +02:00
parent 52c2e31a90
commit 58a06698f0
2 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nodes/image_nodes.py
View File

@ -1731,7 +1731,7 @@ Returns a range of latents from a batch.
elif len(samples.shape) == 5:
chosen_latents = samples[:, :, start_index:end_index]
return ({"samples": chosen_latents,},)
return ({"samples": chosen_latents.contiguous(),},)
class InsertLatentToIndex:

12
nodes/model_optimization_nodes.py
View File

@ -1189,18 +1189,14 @@ class CFGZeroStarAndInit:
def patch(self, model, use_zero_init, zero_star_steps):
def cfg_zerostar(args):
cond = args["cond"]
uncond = args["uncond"]
cond_scale = args["cond_scale"]
#zero init
timestep = args["timestep"]
sigmas = args["model_options"]["transformer_options"]["sample_sigmas"]
matched_step_index = (sigmas == timestep[0] ).nonzero()
if len(matched_step_index) > 0:
current_step_index = matched_step_index.item()
else:
for i in range(len(sigmas) - 1):
# walk from beginning of steps until crossing the timestep
if (sigmas[i] - timestep[0]) * (sigmas[i + 1] - timestep[0]) <= 0:
current_step_index = i
break
@ -1208,7 +1204,11 @@ class CFGZeroStarAndInit:
current_step_index = 0
if (current_step_index <= zero_star_steps) and use_zero_init:
return cond * 0
return cond * 0
cond = args["cond"]
uncond = args["uncond"]
cond_scale = args["cond_scale"]
batch_size = cond.shape[0]