Fixing app.py

app.py

@@ -44,14 +44,15 @@ def infer(prompt, seed=42, randomize_seed=False, aspect_ratio="4:3 landscape 115
     # pipe.load_lora_weights(weight_path)
     # pipe.fuse_lora()
     torch.cuda.empty_cache()
-    image = python.generate_image(
+    image, seed = python.generate_image(
         prompt,
         height,
         width,
+        guidance_scale,
         aspect_ratio,
         seed,
-        guidance_scale=guidance_scale,
-    ).images[0]
+        num_inference_steps,
+    )
     # Generate images
     # for img in pipe.flux_pipe_call_that_returns_an_iterable_of_images(
     #         prompt=prompt,

comfy/k_diffusion/sampling.py

@@ -737,7 +737,7 @@ def sample_dpmpp_2m(model, x, sigmas, extra_args=None, callback=None, disable=No
     t_fn = lambda sigma: sigma.log().neg()
     old_denoised = None
 
-    for i in trange(len(sigmas) - 1, disable=disable):
+    for i in range(len(sigmas) - 1):
         denoised = model(x, sigmas[i] * s_in, **extra_args)
         if callback is not None:
             callback({'x': x, 'i': i, 'sigma': sigmas[i], 'sigma_hat': sigmas[i], 'denoised': denoised})