Update app.py

app.py

@@ -33,21 +33,21 @@ def infer(prompt, seed, randomize_seed):
         seed = random.randint(0, MAX_SEED)
 
     generator = torch.Generator().manual_seed(seed)
-    sigmas = [1.0000, 0.9454, 0.8959, 0.7904, 0.7371, 0.6022]
+    sigmas = [1.0000, 0.9454, 0.8959, 0.7904, 0.7371, 0.6022, 0.0000]
     scales = [32, 48, 64, 80, 96, 128]
-
+    
     images = run(
-        pipe,
-        prompt,
-        sigmas=sigmas,
-        scales=scales,
-        num_inference_steps=6,
-
-        guidance_scale=0.0,
-        height=int(scales[0] * 8),
-        width=int(scales[0] * 8),
-        generator=generator,
-    ).images[0]
+            pipe,
+            prompt,
+            sigmas=torch.tensor(sigmas).to('cuda'),
+            timesteps=torch.tensor(sigmas[:-1]).to('cuda') * 1000,
+            scales=scales,
+        
+            guidance_scale=0.0,
+            height=int(scales[0] * 8),
+            width=int(scales[0] * 8),
+            generator=generator,
+        ).images
 
     return images