diff options
author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-08-06 17:53:33 +0300 |
---|---|---|
committer | AUTOMATIC1111 <16777216c@gmail.com> | 2023-08-06 17:53:33 +0300 |
commit | 5a0db84b6c7322082c7532df11a29a95a59a612b (patch) | |
tree | d28c4ed2cc29a105bcc1895b3600fc24f9c0aaec /modules/processing.py | |
parent | 956e69bf3a9387b6a6cda6823d729e3d3f13c3e1 (diff) |
add infotext
add proper support for recalculating conds in k-diffusion samplers
remove support for compvis samplers
Diffstat (limited to 'modules/processing.py')
-rw-r--r-- | modules/processing.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/modules/processing.py b/modules/processing.py index f4748d6d..ec66fd8e 100644 --- a/modules/processing.py +++ b/modules/processing.py @@ -370,6 +370,9 @@ class StableDiffusionProcessing: self.uc = self.get_conds_with_caching(prompt_parser.get_learned_conditioning, negative_prompts, self.steps * self.step_multiplier, [self.cached_uc], self.extra_network_data)
self.c = self.get_conds_with_caching(prompt_parser.get_multicond_learned_conditioning, prompts, self.steps * self.step_multiplier, [self.cached_c], self.extra_network_data)
+ def get_conds(self):
+ return self.c, self.uc
+
def parse_extra_network_prompts(self):
self.prompts, self.extra_network_data = extra_networks.parse_prompts(self.prompts)
@@ -1251,6 +1254,13 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing): with devices.autocast():
extra_networks.activate(self, self.extra_network_data)
+ def get_conds(self):
+ if self.is_hr_pass:
+ return self.hr_c, self.hr_uc
+
+ return super().get_conds()
+
+
def parse_extra_network_prompts(self):
res = super().parse_extra_network_prompts()
|