soaska/my-sd
soaska/my-sd
1
0
Fork 0
Code Issues Pull Requests Actions Packages Activity

fix ruff

Browse Source
This commit is contained in:
lambertae 2023-07-20 20:36:40 -04:00
parent 2f57a559ac
commit 128d59c9cc
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/sd_samplers_kdiffusion.py
View File

@ -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)