Hotfix #708... (#710)

adieyal · Jan 16, 2024 · 7499182 · 7499182
1 parent eea2ebb
commit 7499182
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git 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
@@ -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}"}]