ford442 commited on
Commit
faccd40
·
verified ·
1 Parent(s): a2f5e03

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -90,8 +90,8 @@ def load_and_prepare_model():
90
  )
91
  #pipe.vae = vaeRV #.to(torch.bfloat16)
92
  pipe.scheduler = sched
93
- pipe.vae.do_resize=False
94
- pipe.vae.vae_scale_factor=8
95
  pipe.to(device=device, dtype=torch.bfloat16)
96
  pipe.vae.set_default_attn_processor()
97
  print(f'init noise scale: {pipe.scheduler.init_noise_sigma}')
 
90
  )
91
  #pipe.vae = vaeRV #.to(torch.bfloat16)
92
  pipe.scheduler = sched
93
+ #pipe.vae.do_resize=False
94
+ #pipe.vae.vae_scale_factor=8
95
  pipe.to(device=device, dtype=torch.bfloat16)
96
  pipe.vae.set_default_attn_processor()
97
  print(f'init noise scale: {pipe.scheduler.init_noise_sigma}')