nyanko7 commited on
Commit
0e71f04
·
verified ·
1 Parent(s): f69d4c3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = 20
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: