diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-08-06 04:41:17 +0000 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2023-08-06 04:41:24 +0000 |
| commit | aa42c0ff8e51c1dde50a313aa2c12b357b287b50 (patch) | |
| tree | d6b3d731b1fa57e323127e27dc2f26484224a230 /modules/sd_samplers_common.py | |
| parent | 06da34d47aef2fbf1326fd3214d4823e06db23b9 (diff) | |
| download | stable-diffusion-webui-gfx803-aa42c0ff8e51c1dde50a313aa2c12b357b287b50.tar.gz stable-diffusion-webui-gfx803-aa42c0ff8e51c1dde50a313aa2c12b357b287b50.tar.bz2 stable-diffusion-webui-gfx803-aa42c0ff8e51c1dde50a313aa2c12b357b287b50.zip | |
repair broken live previews if using VAE with half
Diffstat (limited to 'modules/sd_samplers_common.py')
| -rw-r--r-- | modules/sd_samplers_common.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/sd_samplers_common.py b/modules/sd_samplers_common.py index 42a29fc9..39586b40 100644 --- a/modules/sd_samplers_common.py +++ b/modules/sd_samplers_common.py @@ -39,7 +39,7 @@ def samples_to_images_tensor(sample, approximation=None, model=None): else:
if model is None:
model = shared.sd_model
- x_sample = model.decode_first_stage(sample)
+ x_sample = model.decode_first_stage(sample.to(model.first_stage_model.dtype))
return x_sample
|