diff --git a/nodes.py b/nodes.py
index 54f2c82..0d56843 100644
--- a/nodes.py
+++ b/nodes.py
@@ -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)