RealVis_v5.0_BF16_E

Runtime error

1inkusFace commited on Jan 20

Commit

f76af3c

verified ·

1 Parent(s): ba2840f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -176,7 +176,7 @@ def load_and_prepare_model():
     #**** BETTER WAY ****#
     pipe.to(device, torch.bfloat16)
     #**** BETTER WAY ****#
-    pipe.vae = vaeX.to(device) #.to(torch.bfloat16)
     pipe.vae.set_default_attn_processor()
     #pipe.to(device)
     #pipe.to(torch.bfloat16)

     #**** BETTER WAY ****#
     pipe.to(device, torch.bfloat16)
     #**** BETTER WAY ****#
+    pipe.vae = vaeXL.to(device) #.to(torch.bfloat16)
     pipe.vae.set_default_attn_processor()
     #pipe.to(device)
     #pipe.to(torch.bfloat16)