Disable Hires checkpoint if same as First pass checkpoint (#16269)

This commit is contained in:
w-e-w 2024-10-29 23:45:45 +09:00 committed by GitHub
parent 4ec10bcdc1
commit 14c6d6cb3a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1259,7 +1259,10 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing):
if self.hr_checkpoint_info is None:
raise Exception(f'Could not find checkpoint with name {self.hr_checkpoint_name}')
self.extra_generation_params["Hires checkpoint"] = self.hr_checkpoint_info.short_title
if shared.sd_model.sd_checkpoint_info == self.hr_checkpoint_info:
self.hr_checkpoint_info = None
else:
self.extra_generation_params["Hires checkpoint"] = self.hr_checkpoint_info.short_title
if self.hr_sampler_name is not None and self.hr_sampler_name != self.sampler_name:
self.extra_generation_params["Hires sampler"] = self.hr_sampler_name