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

Refactor code to iterate over cfg and seed in common_ksampler_xyz()

Browse Source
This commit is contained in:
Ayberk Aksoy
2024-01-01 15:01:18 -05:00
parent cec8c759be
commit f7c1f50bed
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
@@ -171,22 +171,23 @@ 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)
for cfg_ in cfg:
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')
cfg_ = cfg_.to('cuda')
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)
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)