Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-18 16:54:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix reprocess workflow for control with hires

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-09-06 11:28:42 -04:00
parent 4f5e066347
commit ff9a7e5444
6 changed files with 33 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_class.py
+8 -6
View File
@@ -368,8 +368,8 @@ class StableDiffusionProcessing:
shared.opts.sd_model_checkpoint = sd_model_checkpoint
sd_models.reload_model_weights()
def __str__(self):
return f'{self.__class__.__name__}: {self.__dict__}'
def __repr__(self):
return f'{self.__class__.__name__}({", ".join([f"{k}={v}" for k, v in self.__dict__.items() if k not in ["scripts_value", "script_args_value"]])})'
@property
def sd_model(self):
@@ -557,15 +557,17 @@ class StableDiffusionProcessingControl(StableDiffusionProcessingImg2Img):
debug(f'Process init: mode={self.__class__.__name__} kwargs={kwargs}') # pylint: disable=protected-access
super().__init__(**kwargs)
def init_hr(self, scale = None, upscaler = None, force = False):
scale = scale or self.scale_by
upscaler = upscaler or self.resize_name
def init_hr(self, scale:float=None, upscaler:str=None, force:bool=False):
scale = scale or self.scale_by or self.scale_by_before
upscaler = upscaler or self.hr_upscaler or self.resize_name or self.resize_name_before
if upscaler is None:
upscaler = 'None'
# self.hr_upscaler = upscaler or 'None'
use_scale = self.hr_resize_x == 0 or self.hr_resize_y == 0
if upscaler == 'None' or (use_scale and scale == 1.0):
return
self.is_hr_pass = True
self.hr_force = force
self.hr_upscaler = upscaler
if use_scale:
vae_scale_factor = sd_vae.get_vae_scale_factor()
self.hr_upscale_to_x, self.hr_upscale_to_y = vae_scale_factor * int(self.width * scale / vae_scale_factor), vae_scale_factor * int(self.height * scale / vae_scale_factor)