Skip to content

Commit 14cf434

Browse files
committed
fix an issue in live previews that happens when you use SDXL with fp16 VAE
1 parent 5dee0fa commit 14cf434

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

modules/processing.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -539,8 +539,7 @@ def create_random_tensors(shape, seeds, subseeds=None, subseed_strength=0.0, see
539539

540540

541541
def decode_first_stage(model, x):
542-
with devices.autocast(disable=x.dtype == devices.dtype_vae):
543-
x = model.decode_first_stage(x)
542+
x = model.decode_first_stage(x.to(devices.dtype_vae))
544543

545544
return x
546545

0 commit comments

Comments
 (0)