fix hires and corrections with batch processing

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2024-12-30 21:31:35 -05:00
parent 86ac38d94f
commit 05d5ac03be
6 changed files with 42 additions and 49 deletions
+7 -6
View File
@@ -116,16 +116,18 @@ class StableDiffusionProcessing:
# overrides
override_settings: Dict[str, Any] = {},
override_settings_restore_afterwards: bool = True,
task_args: Dict[str, Any] = {},
ops: List[str] = [],
# metadata
extra_generation_params: Dict[Any, Any] = {},
):
# extra args set by processing loop
self.task_args = {}
self.task_args = task_args
# state items
self.state: str = ''
self.ops = []
self.ops = ops
self.skip = []
self.color_corrections = []
self.is_control = False
@@ -266,7 +268,6 @@ class StableDiffusionProcessing:
self.s_max = shared.opts.s_max
self.s_tmin = shared.opts.s_tmin
self.s_tmax = float('inf') # not representable as a standard ui option
self.task_args = {}
# ip adapter
self.ip_adapter_names = []
@@ -299,6 +300,9 @@ class StableDiffusionProcessing:
self.negative_embeds = []
self.negative_pooleds = []
def __str__(self):
return f'{self.__class__.__name__}: {self.__dict__}'
@property
def sd_model(self):
return shared.sd_model
@@ -339,9 +343,6 @@ class StableDiffusionProcessing:
def close(self):
self.sampler = None # pylint: disable=attribute-defined-outside-init
def __str__(self):
return f'{self.__class__.__name__}: {self.__dict__}'
class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing):
def __init__(self, **kwargs):