Merge pull request #96 from pythias/multi-prompts

support multi prompts
ototadana · Jun 16, 2023 · 601efd0 · 601efd0
2 parents 9ddeed8 + eb6a1f9
commit 601efd0
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/scripts/use_cases/image_processor.py b/scripts/use_cases/image_processor.py
@@ -52,6 +52,10 @@ def proc_images(self, o: StableDiffusionProcessing, res: Processed, option: Opti
             if subseed_index < len(res.all_subseeds):
                 p.subseed = res.all_subseeds[subseed_index]
                 subseed_index += 1
+
+            if type(p.prompt) == list:
+                p.prompt = p.prompt[i]
+
             proc = self.proc_image(p, option, image)
             edited_images.extend(proc.images)
             all_seeds.extend(proc.all_seeds)
@@ -79,9 +83,7 @@ def __init_processing(self, p: StableDiffusionProcessingImg2Img, o: StableDiffus
         p.width, p.height = image.size
         p.sample = sample
 
-    def proc_image(
-        self, p: StableDiffusionProcessingImg2Img, option: Option, pre_proc_image: Image = None
-    ) -> Processed:
+    def proc_image(self, p: StableDiffusionProcessingImg2Img, option: Option, pre_proc_image: Image = None) -> Processed:
         params = option.to_dict()
 
         if hasattr(p.init_images[0], "mode") and p.init_images[0].mode != "RGB":