Commit 1c9d1b0e authored by w-e-w's avatar w-e-w

simplify self.extra_network_data

parent f098e726
...@@ -339,9 +339,7 @@ class StableDiffusionProcessing: ...@@ -339,9 +339,7 @@ class StableDiffusionProcessing:
self.c = self.get_conds_with_caching(prompt_parser.get_multicond_learned_conditioning, self.prompts, self.steps * self.step_multiplier, self.cached_c, self.extra_network_data) self.c = self.get_conds_with_caching(prompt_parser.get_multicond_learned_conditioning, self.prompts, self.steps * self.step_multiplier, self.cached_c, self.extra_network_data)
def parse_extra_network_prompts(self): def parse_extra_network_prompts(self):
self.prompts, extra_network_data = extra_networks.parse_prompts(self.prompts) self.prompts, self.extra_network_data = extra_networks.parse_prompts(self.prompts)
return extra_network_data
class Processed: class Processed:
...@@ -702,7 +700,7 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed: ...@@ -702,7 +700,7 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
if len(p.prompts) == 0: if len(p.prompts) == 0:
break break
p.extra_network_data = p.parse_extra_network_prompts() p.parse_extra_network_prompts()
if not p.disable_extra_networks: if not p.disable_extra_networks:
with devices.autocast(): with devices.autocast():
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment