Skip to content

GitLab

"...resnet50_tensorflow.git" did not exist on "d3599b2ea4998b139cca1c4681f44d8431b39c24"
Commit 5d875d77 authored by comfyanonymous's avatar comfyanonymous
Browse files

Fix regression with lcm not working with batches.

parent 4b9005e9
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
comfy/k_diffusion/sampling.py
View file @ 5d875d77
...@@ -748,7 +748,7 @@ def sample_lcm(model, x, sigmas, extra_args=None, callback=None, disable=None, n ...@@ -748,7 +748,7 @@ def sample_lcm(model, x, sigmas, extra_args=None, callback=None, disable=None, n
x = denoised x = denoised
if sigmas[i + 1] > 0: if sigmas[i + 1] > 0:
x = model.inner_model.inner_model.model_sampling.noise_scaling(sigmas[i + 1] * s_in, noise_sampler(sigmas[i], sigmas[i + 1]), x) x = model.inner_model.inner_model.model_sampling.noise_scaling(sigmas[i + 1], noise_sampler(sigmas[i], sigmas[i + 1]), x)
return x return x
......
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