aboutsummaryrefslogtreecommitdiff
path: root/modules/sd_samplers.py
diff options
context:
space:
mode:
authorrandom_thoughtss <random_thoughtss@proton.me>2022-10-19 15:09:43 -0700
committerrandom_thoughtss <random_thoughtss@proton.me>2022-10-19 15:09:43 -0700
commitc418467c03db916c3e5312e6ac4a67365e196dbd (patch)
tree42d4a5a2219112c34d47d5419e8ce939450be5bc /modules/sd_samplers.py
parentdde9f960727bfe151d418e43685a2881cf580a17 (diff)
Don't compute latent mask if were not using it. Also added support for fixed highres_fix generation.
Diffstat (limited to 'modules/sd_samplers.py')
-rw-r--r--modules/sd_samplers.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/modules/sd_samplers.py b/modules/sd_samplers.py
index d270e4df..c21be26e 100644
--- a/modules/sd_samplers.py
+++ b/modules/sd_samplers.py
@@ -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"]