nodes/CYBERLOOM-INC-ComfyUI-nodes-hnmr
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Refactor common_ksampler_xyz function to fix multiple seed input issue

Browse Source
This commit is contained in:
Ayberk Aksoy
2023-12-31 19:31:19 -05:00
parent 183ded8d4e
commit cec8c759be
1 changed files with 16 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
sample.py
+7 -6
View File
@@ -157,11 +157,6 @@ def common_ksampler_xyz(
latent_image = latent["samples"]
noise_mask = latent.get('noise_mask', None)
noise, latent_image = get_noise(seed, latent_image, disable_noise, latent.get('batch_index', 0))
noise, latent_image, cfg_ = get_cfg(noise, latent_image, cfg)
cfg_ = cfg_.to('cuda')
all_samples: List[torch.Tensor] = []
for (
model_index, model_fn, step, sampler, scheduler
@@ -176,9 +171,15 @@ def common_ksampler_xyz(
if alphas is not None:
print(f'alpha = {alphas}')
for seed_ in seed:
noise, latent_image = get_noise([seed_], latent_image, disable_noise, latent.get('batch_index', 0))
noise, latent_image, cfg_ = get_cfg(noise, latent_image, cfg)
cfg_ = cfg_.to('cuda')
samples = comfy.sample.sample(
current_model, noise, step, cfg_, sampler, scheduler,
positive_copy, negative_copy, latent_image,
positive_copy, negative_copy, latent_image[0],
denoise=denoise, disable_noise=disable_noise,
start_step=start_step, last_step=last_step,
force_full_denoise=force_full_denoise, noise_mask=noise_mask