From 671adb6dba2d0252cb27d3721aff113769adfaf7 Mon Sep 17 00:00:00 2001
From: Om Duggineni <omduggineni@gmail.com>
Date: Thu, 30 Jun 2022 18:53:20 -0400
Subject: [PATCH] hack: the server should always return num_images images

the number of images returned no longer depends on the number of GPUs
---
 executors/dalle/executor.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/executors/dalle/executor.py b/executors/dalle/executor.py
index c3ad1ec..beef793 100644
--- a/executors/dalle/executor.py
+++ b/executors/dalle/executor.py
@@ -16,7 +16,7 @@ def generate(self, docs: DocumentArray, parameters: Dict, **kwargs):
         request_time = time.time()
         for d in docs:
             self.logger.info(f'dalle {num_images} [{d.text}]...')
-            generated_imgs = dm_helper.generate_images(d.text, num_images)
+            generated_imgs = dm_helper.generate_images(d.text, num_images)[:num_images]
 
             for img in generated_imgs:
                 buffered = BytesIO()