Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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
|