device

app.py

@@ -258,7 +258,10 @@ class VideoCoF_Controller(Wan_Controller):
             torch.manual_seed(int(seed_textbox))
         else: 
             seed_textbox = np.random.randint(0, 1e10)
-        generator = torch.Generator(device=self.device).manual_seed(int(seed_textbox))
+        # Ensure generator is created on the same device as the pipeline's transformer
+        gen_device = getattr(getattr(self, "pipeline", None), "transformer", None)
+        gen_device = gen_device.device if gen_device is not None else self.device
+        generator = torch.Generator(device=gen_device).manual_seed(int(seed_textbox))
 
         try:
             # VideoCoF logic