envs

app.py

@@ -286,7 +286,7 @@ class ImageConductor:
         image_norm = lambda x: x
         image_paths = [first_frame_path]
         controlnet_images = [image_norm(image_transforms(Image.open(path).convert("RGB"))) for path in image_paths]
-        controlnet_images = torch.stack(controlnet_images).unsqueeze(0).cuda()
+        controlnet_images = torch.stack(controlnet_images).unsqueeze(0).to(device)
         controlnet_images = rearrange(controlnet_images, "b f c h w -> b c f h w")
         num_controlnet_images = controlnet_images.shape[2]
         controlnet_images = rearrange(controlnet_images, "b c f h w -> (b f) c h w")