RealVis_v5.0_BF16_IP_B

Running on Zero

1inkusFace commited on 20 days ago

Commit

d687be5

verified ·

1 Parent(s): f81797f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -140,7 +140,7 @@ def load_and_prepare_model():
     '''
     #pipe.vae=vaeX
     pipe.to(device=device, dtype=torch.bfloat16)
-    pipe.vae=vaeX
    # pipe.vae.enable_tiling()
    # pipe.vae.fuse_qkv_projections()
     #pipe.vae.to(device=device, dtype=torch.bfloat16)

     '''
     #pipe.vae=vaeX
     pipe.to(device=device, dtype=torch.bfloat16)
+    pipe.vae=vaeX.to(device)
    # pipe.vae.enable_tiling()
    # pipe.vae.fuse_qkv_projections()
     #pipe.vae.to(device=device, dtype=torch.bfloat16)