diff options
| author | Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com> | 2023-08-04 06:38:52 +0000 |
|---|---|---|
| committer | Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com> | 2023-08-04 06:38:52 +0000 |
| commit | 1f6bfdea80f58f292aeebb9a001689a118d71c01 (patch) | |
| tree | 3a73b11529e9a782b0a6715d2039a98d989fac38 | |
| parent | 70e66e81e56f0bb187395878ee49705acbfab40c (diff) | |
| download | stable-diffusion-webui-gfx803-1f6bfdea80f58f292aeebb9a001689a118d71c01.tar.gz stable-diffusion-webui-gfx803-1f6bfdea80f58f292aeebb9a001689a118d71c01.tar.bz2 stable-diffusion-webui-gfx803-1f6bfdea80f58f292aeebb9a001689a118d71c01.zip | |
move the modified decode into smapler_common
| -rw-r--r-- | modules/sd_samplers_common.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/modules/sd_samplers_common.py b/modules/sd_samplers_common.py index 2cfa4ac6..7269514f 100644 --- a/modules/sd_samplers_common.py +++ b/modules/sd_samplers_common.py @@ -55,9 +55,9 @@ def single_sample_to_image(sample, approximation=None): def decode_first_stage(model, x):
- x = model.decode_first_stage(x.to(devices.dtype_vae))
-
- return x
+ x = x.to(devices.dtype_vae)
+ approx_index = approximation_indexes.get(opts.sd_vae_decode_method, 0)
+ return samples_to_images_tensor(x, approx_index, model)
def sample_to_image(samples, index=0, approximation=None):
|