when doing --log_tokenization/-t also log parsed prompt

This commit is contained in:
Damian Stewart 2022-11-21 19:27:40 +01:00
parent ce8c2bea2f
commit e0581a2c37

View File

@ -47,6 +47,9 @@ def get_uc_and_c_and_ec(prompt_string_uncleaned, model, log_tokens=False, skip_n
parsed_prompt = pp.parse_conjunction(prompt_string_cleaned).prompts[0]
parsed_negative_prompt: FlattenedPrompt = pp.parse_conjunction(unconditioned_words).prompts[0]
if log_tokens:
print(f">> Parsed prompt to {parsed_prompt}")
print(f">> Parsed negative prompt to {parsed_negative_prompt}")
conditioning = None
cac_args:cross_attention_control.Arguments = None