Spaces:
Running
on
Zero
Running
on
Zero
envs
Browse files
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).
|
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")
|