added support for AND from https://energy-based-model.github.io/Compositional-Visual-Generation-with-Composable-Diffusion-Models/

1 files changed, 4 insertions, 2 deletions

diff --git a/modules/ui.py b/modules/ui.py
index 523ab25b..9620350f 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -34,7 +34,7 @@ import modules.gfpgan_model
 import modules.codeformer_model

 import modules.styles

 import modules.generation_parameters_copypaste

-from modules.prompt_parser import get_learned_conditioning_prompt_schedules

+from modules import prompt_parser

 from modules.images import apply_filename_pattern, get_next_sequence_number

 import modules.textual_inversion.ui

 

@@ -394,7 +394,9 @@ def connect_reuse_seed(seed: gr.Number, reuse_seed: gr.Button, generation_info:
 

 def update_token_counter(text, steps):

     try:

-        prompt_schedules = get_learned_conditioning_prompt_schedules([text], steps)

+        _, prompt_flat_list, _ = prompt_parser.get_multicond_prompt_list([text])

+        prompt_schedules = prompt_parser.get_learned_conditioning_prompt_schedules(prompt_flat_list, steps)

+

     except Exception:

         # a parsing error can happen here during typing, and we don't want to bother the user with

         # messages related to it in console