repair broken live previews if using VAE with half

2023-08-06 07:41:17 +03:00 · 2023-08-06 07:41:17 +03:00 · aa42c0ff8e
parent 06da34d47a
commit aa42c0ff8e
1 changed files with 1 additions and 1 deletions
--- 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