From 7499182064080a062ec2b68ceea3d19c121fef12 Mon Sep 17 00:00:00 2001 From: Aarni Koskela Date: Tue, 16 Jan 2024 13:18:51 +0200 Subject: [PATCH] Hotfix #708... (#710) --- sd_dynamic_prompts/magic_prompt.py | 3 ++- tests/prompts/test_magic_prompts.py | 1 + 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/sd_dynamic_prompts/magic_prompt.py b/sd_dynamic_prompts/magic_prompt.py index 40d82fba..7dcc3e83 100644 --- a/sd_dynamic_prompts/magic_prompt.py +++ b/sd_dynamic_prompts/magic_prompt.py @@ -17,7 +17,8 @@ def _generate_magic_prompts(self, orig_prompts: list[str]) -> list[str]: orig_prompts, chunks = zip( *(remove_a1111_special_syntax_chunks(p) for p in orig_prompts), ) - magic_prompts = super()._generate_magic_prompts(orig_prompts) + # `transformers` is rather particular that the input is a list, not a tuple + magic_prompts = super()._generate_magic_prompts(list(orig_prompts)) # in case we somehow get less magic prompts than we started with, # use zip_longest instead of zip. return [ diff --git a/tests/prompts/test_magic_prompts.py b/tests/prompts/test_magic_prompts.py index d32cc653..f8445119 100644 --- a/tests/prompts/test_magic_prompts.py +++ b/tests/prompts/test_magic_prompts.py @@ -1,4 +1,5 @@ def fake_generator(prompts, **_kwargs): + assert isinstance(prompts, list) # be as particular as transformers is for prompt in prompts: assert "<" not in prompt # should have been stripped yield [{"generated_text": f"magical {prompt}"}]