diff options
| author | catboxanon <122327233+catboxanon@users.noreply.github.com> | 2023-11-29 18:10:43 -0500 |
|---|---|---|
| committer | catboxanon <122327233+catboxanon@users.noreply.github.com> | 2023-11-29 18:10:43 -0500 |
| commit | ffa7f8201d849636bb327b3b40298e7c169ff204 (patch) | |
| tree | 76ebfecf53f2135d0658588b55ba7295043a86da | |
| parent | ec6ee5c13bf3453f8703e225a191333a9bbcf10a (diff) | |
Lint
| -rw-r--r-- | modules/processing.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/modules/processing.py b/modules/processing.py index c88eec70..f3883d5b 100644 --- a/modules/processing.py +++ b/modules/processing.py @@ -863,7 +863,7 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed: if p.n_iter > 1:
shared.state.job = f"Batch {n+1} out of {p.n_iter}"
-
+
def rescale_zero_terminal_snr_abar(alphas_cumprod):
alphas_bar_sqrt = alphas_cumprod.sqrt()
@@ -881,7 +881,7 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed: alphas_bar = alphas_bar_sqrt**2 # Revert sqrt
alphas_bar[-1] = 4.8973451890853435e-08
return alphas_bar
-
+
p.sd_model.alphas_cumprod = p.sd_model.alphas_cumprod_original.to(shared.device)
if opts.use_downcasted_alpha_bar:
|