Rebuild guidance_schedule to match step count

app.py

@@ -37,16 +37,21 @@ def generate(
         seed = random.randint(0, MAX_SEED)
     generator = torch.Generator(device="cuda").manual_seed(int(seed))
 
+    steps = int(num_inference_steps)
     kwargs = dict(
         prompt=prompt,
         width=int(width),
         height=int(height),
-        num_inference_steps=int(num_inference_steps),
+        num_inference_steps=steps,
         generator=generator,
     )
     if guidance_scale > 0:
         kwargs["guidance_scale"] = float(guidance_scale)
         kwargs["guidance_schedule"] = None
+    else:
+        # PR default is len 48 (7.0 x45 + 3.0 x3); rebuild it for any step count.
+        tail = min(3, max(0, steps - 1))
+        kwargs["guidance_schedule"] = (7.0,) * (steps - tail) + (3.0,) * tail
 
     image = pipe(**kwargs).images[0]
     return image, seed