mirror of
https://github.com/openvinotoolkit/stable-diffusion-webui.git
synced 2024-12-14 14:45:06 +03:00
fix an issue in live previews that happens when you use SDXL with fp16 VAE
This commit is contained in:
parent
5dee0fa1f8
commit
14cf434bc3
@ -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
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user