Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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}')
|