batch image processing use captions

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2024-12-21 10:16:33 -05:00
parent 76755c6b6e
commit 29c76cd8fa
5 changed files with 70 additions and 50 deletions
+5 -3
View File
@@ -348,13 +348,15 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing):
debug(f'Process init: mode={self.__class__.__name__} kwargs={kwargs}') # pylint: disable=protected-access
super().__init__(**kwargs)
def init(self, all_prompts=None, all_seeds=None, all_subseeds=None):
def init(self, all_prompts=None, all_negative_prompts=None, all_seeds=None, all_subseeds=None):
if shared.native:
shared.sd_model = sd_models.set_diffuser_pipe(self.sd_model, sd_models.DiffusersTaskType.TEXT_2_IMAGE)
self.width = self.width or 512
self.height = self.height or 512
self.width = self.width or 1024
self.height = self.height or 1024
if all_prompts is not None:
self.all_prompts = all_prompts
if all_negative_prompts is not None:
self.all_negative_prompts = all_negative_prompts
if all_seeds is not None:
self.all_seeds = all_seeds
if all_subseeds is not None: