Yw22 commited on
Commit
2d84a5c
1 Parent(s): 33be0eb
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -286,7 +286,7 @@ class ImageConductor:
286
  image_norm = lambda x: x
287
  image_paths = [first_frame_path]
288
  controlnet_images = [image_norm(image_transforms(Image.open(path).convert("RGB"))) for path in image_paths]
289
- controlnet_images = torch.stack(controlnet_images).unsqueeze(0).cuda()
290
  controlnet_images = rearrange(controlnet_images, "b f c h w -> b c f h w")
291
  num_controlnet_images = controlnet_images.shape[2]
292
  controlnet_images = rearrange(controlnet_images, "b c f h w -> (b f) c h w")
 
286
  image_norm = lambda x: x
287
  image_paths = [first_frame_path]
288
  controlnet_images = [image_norm(image_transforms(Image.open(path).convert("RGB"))) for path in image_paths]
289
+ controlnet_images = torch.stack(controlnet_images).unsqueeze(0).to(device)
290
  controlnet_images = rearrange(controlnet_images, "b f c h w -> b c f h w")
291
  num_controlnet_images = controlnet_images.shape[2]
292
  controlnet_images = rearrange(controlnet_images, "b c f h w -> (b f) c h w")