1inkusFace commited on
Commit
a09af35
·
verified ·
1 Parent(s): 902fb5e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -121,7 +121,7 @@ def load_and_prepare_model():
121
  text_encoder=None,
122
  text_encoder_2=None,
123
  vae=None,
124
- unet=None,
125
  )
126
 
127
  '''
@@ -141,8 +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.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
 
121
  text_encoder=None,
122
  text_encoder_2=None,
123
  vae=None,
124
+ # unet=None,
125
  )
126
 
127
  '''
 
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