Commit aa42c0ff authored by AUTOMATIC1111's avatar AUTOMATIC1111
Browse files

repair broken live previews if using VAE with half

parent 06da34d4
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -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