flowedit fix

2025-07-31 12:21:40 +03:00
parent e545b18627
commit 0cc4583653
1 changed files with 1 additions and 0 deletions
@@ -1873,6 +1873,7 @@ class WanVideoSampler:
        # FlowEdit setup
        if flowedit_args is not None:
            source_embeds = flowedit_args["source_embeds"]
+            source_embeds = dict_to_device(source_embeds, device)
            source_image_embeds = flowedit_args.get("source_image_embeds", image_embeds)
            source_image_cond = source_image_embeds.get("image_embeds", None)
            source_clip_fea = source_image_embeds.get("clip_fea", clip_fea)