structure

routers/intference/stable_diffusion.py

@@ -48,7 +48,7 @@ async def generate(prompt: str):
         num_inference_steps=n_steps,
         denoising_end=high_noise_frac,
         output_type="latent",
-    ).images
+    ).images[0]
     final_image = refiner(
         prompt=prompt,
         negative_prompt=negative,