Don't compute latent mask if were not using it. Also added support for fixed highres_fix generation.

This commit is contained in:
random_thoughtss
2022-10-19 15:09:43 -07:00
parent dde9f96072
commit c418467c03
2 changed files with 48 additions and 26 deletions

View File

@@ -117,6 +117,8 @@ class VanillaStableDiffusionSampler:
self.config = None
self.last_latent = None
self.conditioning_key = sd_model.model.conditioning_key
def number_of_needed_noises(self, p):
return 0
@@ -328,6 +330,8 @@ class KDiffusionSampler:
self.config = None
self.last_latent = None
self.conditioning_key = sd_model.model.conditioning_key
def callback_state(self, d):
step = d['i']
latent = d["denoised"]