ford442 commited on
Commit
199b7ab
·
verified ·
1 Parent(s): f5a68ef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -138,7 +138,7 @@ def load_and_prepare_model():
138
  #pipe.vae.do_resize=False
139
  #pipe.vae.vae_scale_factor=8
140
  pipe.to(device=device, dtype=torch.bfloat16)
141
- pipe.vae.set_default_attn_processor()
142
  print(f'init noise scale: {pipe.scheduler.init_noise_sigma}')
143
  pipe.watermark=None
144
  pipe.safety_checker=None
 
138
  #pipe.vae.do_resize=False
139
  #pipe.vae.vae_scale_factor=8
140
  pipe.to(device=device, dtype=torch.bfloat16)
141
+ #pipe.vae.set_default_attn_processor()
142
  print(f'init noise scale: {pipe.scheduler.init_noise_sigma}')
143
  pipe.watermark=None
144
  pipe.safety_checker=None