Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -30,7 +30,7 @@ noise_scheduler = DDIMScheduler(
|
|
| 30 |
set_alpha_to_one=False,
|
| 31 |
steps_offset=1,
|
| 32 |
)
|
| 33 |
-
vae = AutoencoderKL.from_pretrained(vae_model_path)
|
| 34 |
|
| 35 |
def image_grid(imgs, rows, cols):
|
| 36 |
assert len(imgs) == rows * cols
|
|
|
|
| 30 |
set_alpha_to_one=False,
|
| 31 |
steps_offset=1,
|
| 32 |
)
|
| 33 |
+
vae = AutoencoderKL.from_pretrained(vae_model_path)#.to(dtype=torch.float16)
|
| 34 |
|
| 35 |
def image_grid(imgs, rows, cols):
|
| 36 |
assert len(imgs) == rows * cols
|