repair broken live previews if using VAE with half

aa42c0ff · AUTOMATIC1111 · 06da34d4 · aa42c0ff
Commit aa42c0ff authored Aug 06, 2023 by AUTOMATIC1111
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

modules/sd_samplers_common.py modules/sd_samplers_common.py +1 -1

No files found.
--- 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