mirror of
https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
synced 2025-08-04 11:12:35 +00:00
fix an issue in live previews that happens when you use SDXL with fp16 VAE
This commit is contained in:
@@ -539,8 +539,7 @@ def create_random_tensors(shape, seeds, subseeds=None, subseed_strength=0.0, see
|
|||||||
|
|
||||||
|
|
||||||
def decode_first_stage(model, x):
|
def decode_first_stage(model, x):
|
||||||
with devices.autocast(disable=x.dtype == devices.dtype_vae):
|
x = model.decode_first_stage(x.to(devices.dtype_vae))
|
||||||
x = model.decode_first_stage(x)
|
|
||||||
|
|
||||||
return x
|
return x
|
||||||
|
|
||||||
|
Reference in New Issue
Block a user