fix an issue in live previews that happens when you use SDXL with fp16 VAE

This commit is contained in:
AUTOMATIC1111 2023-07-15 07:33:16 +03:00
parent 5dee0fa1f8
commit 14cf434bc3

View File

@ -539,8 +539,7 @@ def create_random_tensors(shape, seeds, subseeds=None, subseed_strength=0.0, see
def decode_first_stage(model, x):
with devices.autocast(disable=x.dtype == devices.dtype_vae):
x = model.decode_first_stage(x)
x = model.decode_first_stage(x.to(devices.dtype_vae))
return x