mhussainahmad
/

sdxl-wombo-optimized

Model card Files Files and versions Community

AlexNijjar commited on Sep 4, 2024

Commit

1cb9eb1

·

unverified ·

1 Parent(s): f18e889

Use first image in list

Files changed (1) hide show

src/pipeline.py +1 -1

src/pipeline.py CHANGED Viewed

@@ -26,4 +26,4 @@ def infer(request: TextToImageRequest, pipeline: StableDiffusionXLPipeline) -> I
         width=request.width,
         height=request.height,
         generator=generator,
-    ).images

         width=request.width,
         height=request.height,
         generator=generator,
+    ).images[0]