fix prompt travel for sdxl (#467)

2024-03-15 03:39:19 -05:00 · 2024-03-15 03:39:19 -05:00 · 6314c17c2f
parent af8d2e3e4f
commit 6314c17c2f
1 changed files with 2 additions and 1 deletions
--- a/scripts/animatediff_prompt.py
+++ b/scripts/animatediff_prompt.py
@ -130,7 +130,8 @@ class AnimateDiffPromptSchedule:
        if isinstance(cond, torch.Tensor):
            return torch.stack(cond_list).to(cond.dtype).to(cond.device)
        else:
-            return {k: torch.stack(v).to(cond[k].dtype).to(cond[k].device) for k, v in cond_list.items()}
+            from modules.prompt_parser import DictWithShape
+            return DictWithShape({k: torch.stack(v).to(cond[k].dtype).to(cond[k].device) for k, v in cond_list.items()}, None)


    @staticmethod