Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -784,7 +784,7 @@ def generate_image(
|
|
784 |
generator = torch.Generator(device=device).manual_seed(seed)
|
785 |
x = torch.randn(1, 16, 2 * math.ceil(height / 16), 2 * math.ceil(width / 16), device=device, dtype=torch.bfloat16, generator=generator)
|
786 |
|
787 |
-
num_steps =
|
788 |
timesteps = get_schedule(num_steps, (x.shape[-1] * x.shape[-2]) // 4, shift=True)
|
789 |
|
790 |
if do_img2img and init_image is not None:
|
|
|
784 |
generator = torch.Generator(device=device).manual_seed(seed)
|
785 |
x = torch.randn(1, 16, 2 * math.ceil(height / 16), 2 * math.ceil(width / 16), device=device, dtype=torch.bfloat16, generator=generator)
|
786 |
|
787 |
+
num_steps = 12
|
788 |
timesteps = get_schedule(num_steps, (x.shape[-1] * x.shape[-2]) // 4, shift=True)
|
789 |
|
790 |
if do_img2img and init_image is not None:
|