RealVis_v5.0_BF16_IP

Running on Zero

ford442 commited on Jan 19, 2025

Commit

7d22cc8

verified ·

1 Parent(s): 2438e6b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -137,7 +137,7 @@ def load_and_prepare_model():
     '''
     pipe.to(device=device, dtype=torch.bfloat16)
-    pipe.vae.to('cpu')
     pipe.vae.do_resize=False
     #pipe.vae.do_rescale=False
     #pipe.vae.do_convert_rgb=True

     '''
     pipe.to(device=device, dtype=torch.bfloat16)
+    pipe.vae.to(device=device, dtype=torch.bfloat16)
     pipe.vae.do_resize=False
     #pipe.vae.do_rescale=False
     #pipe.vae.do_convert_rgb=True