diff options
author | lambertae <dengm@mit.edu> | 2023-07-21 00:36:40 +0000 |
---|---|---|
committer | lambertae <dengm@mit.edu> | 2023-07-21 00:36:40 +0000 |
commit | 128d59c9ccfbc9c7fccd6f1b2fe58bbbb18459f9 (patch) | |
tree | 0f2e040814288f8c49e6b239329b7107e8c522d2 | |
parent | 2f57a559ac3381c1ef2516655c3a3d1088191c54 (diff) | |
download | stable-diffusion-webui-gfx803-128d59c9ccfbc9c7fccd6f1b2fe58bbbb18459f9.tar.gz stable-diffusion-webui-gfx803-128d59c9ccfbc9c7fccd6f1b2fe58bbbb18459f9.tar.bz2 stable-diffusion-webui-gfx803-128d59c9ccfbc9c7fccd6f1b2fe58bbbb18459f9.zip |
fix ruff
-rw-r--r-- | modules/sd_samplers_kdiffusion.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/sd_samplers_kdiffusion.py b/modules/sd_samplers_kdiffusion.py index c72d01c8..21b347ed 100644 --- a/modules/sd_samplers_kdiffusion.py +++ b/modules/sd_samplers_kdiffusion.py @@ -43,7 +43,7 @@ def restart_sampler(model, x, sigmas, extra_args=None, callback=None, disable=No extra_args = {} if extra_args is None else extra_args
s_in = x.new_ones([x.shape[0]])
step_id = 0
- from k_diffusion.sampling import to_d, append_zero, get_sigmas_karras
+ from k_diffusion.sampling import to_d, get_sigmas_karras
def heun_step(x, old_sigma, new_sigma, second_order = True):
nonlocal step_id
denoised = model(x, old_sigma * s_in, **extra_args)
|