1inkusFace commited on
Commit
8415147
·
verified ·
1 Parent(s): f2f7398

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -141,6 +141,8 @@ def load_and_prepare_model():
141
  #pipe.vae=vaeX
142
  pipe.to(device=device, dtype=torch.bfloat16)
143
  pipe.vae=vaeX
 
 
144
  #pipe.vae.to(device=device, dtype=torch.bfloat16)
145
  #pipe.vae.do_resize=False
146
  #pipe.vae.do_rescale=False
 
141
  #pipe.vae=vaeX
142
  pipe.to(device=device, dtype=torch.bfloat16)
143
  pipe.vae=vaeX
144
+ pipe.vae.enable_tiling()
145
+ pipe.vae.fuse_qkv_projections()
146
  #pipe.vae.to(device=device, dtype=torch.bfloat16)
147
  #pipe.vae.do_resize=False
148
  #pipe.vae.do_rescale=False