Skip to content

GitLab

Commit f03dade5 authored by comfyanonymous's avatar comfyanonymous
Browse files

Fix bug.

parent 79d08997
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
nodes.py
View file @ f03dade5
...@@ -708,8 +708,9 @@ def common_ksampler(model, seed, steps, cfg, sampler_name, scheduler, positive, ...@@ -708,8 +708,9 @@ def common_ksampler(model, seed, steps, cfg, sampler_name, scheduler, positive,
batch_index = latent["batch_index"] batch_index = latent["batch_index"]
generator = torch.manual_seed(seed) generator = torch.manual_seed(seed)
for i in range(batch_index + 1): for i in range(batch_index):
noise = torch.randn([1] + list(latent_image.size())[1:], dtype=latent_image.dtype, layout=latent_image.layout, generator=generator, device="cpu") noise = torch.randn([1] + list(latent_image.size())[1:], dtype=latent_image.dtype, layout=latent_image.layout, generator=generator, device="cpu")
noise = torch.randn(latent_image.size(), dtype=latent_image.dtype, layout=latent_image.layout, generator=generator, device="cpu")
if "noise_mask" in latent: if "noise_mask" in latent:
noise_mask = latent['noise_mask'] noise_mask = latent['noise_mask']
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment