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
+16 -15
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
@@ -175,17 +170,23 @@ def common_ksampler_xyz(
alphas = merge2.get_current_alpha(current_model.model)
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,
denoise=denoise, disable_noise=disable_noise,
start_step=start_step, last_step=last_step,
force_full_denoise=force_full_denoise, noise_mask=noise_mask
)
samples = samples.cpu()
all_samples.append(samples)
samples = comfy.sample.sample(
current_model, noise, step, cfg_, sampler, scheduler,
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
)
samples = samples.cpu()
all_samples.append(samples)
out = latent.copy()
out["samples"] = torch.cat(all_samples)