aboutsummaryrefslogtreecommitdiffstats
path: root/modules/sd_samplers_extra.py
diff options
context:
space:
mode:
authorKohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com>2023-11-26 09:32:52 +0000
committerKohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com>2023-11-26 09:32:52 +0000
commit3d341ebc7dcb44df3b4c013b3805c08d8a35e24a (patch)
tree3152f9087cdb2a9b48310aca3189d8cc235ffa51 /modules/sd_samplers_extra.py
parent40ac134c553ac824d4a96666bba14d550300daa5 (diff)
parente44103264d95a2141410571661c757a65f22a8b6 (diff)
downloadstable-diffusion-webui-gfx803-3d341ebc7dcb44df3b4c013b3805c08d8a35e24a.tar.gz
stable-diffusion-webui-gfx803-3d341ebc7dcb44df3b4c013b3805c08d8a35e24a.tar.bz2
stable-diffusion-webui-gfx803-3d341ebc7dcb44df3b4c013b3805c08d8a35e24a.zip
Merge branch 'dev' into test-fp8
Diffstat (limited to 'modules/sd_samplers_extra.py')
-rw-r--r--modules/sd_samplers_extra.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/sd_samplers_extra.py b/modules/sd_samplers_extra.py
index 1b981ca8..72fd0aa5 100644
--- a/modules/sd_samplers_extra.py
+++ b/modules/sd_samplers_extra.py
@@ -60,7 +60,7 @@ def restart_sampler(model, x, sigmas, extra_args=None, callback=None, disable=No
sigma_restart = get_sigmas_karras(restart_steps, sigmas[min_idx].item(), sigmas[max_idx].item(), device=sigmas.device)[:-1]
while restart_times > 0:
restart_times -= 1
- step_list.extend([(old_sigma, new_sigma) for (old_sigma, new_sigma) in zip(sigma_restart[:-1], sigma_restart[1:])])
+ step_list.extend(zip(sigma_restart[:-1], sigma_restart[1:]))
last_sigma = None
for old_sigma, new_sigma in tqdm.tqdm(step_list, disable=disable):