Stable-Video-Diffusion-Img2Vid

Running on Zero

Fabrice-TIERCELIN commited on Jul 15

Commit

ce08569

•

1 Parent(s): df89211

Fully qualified name

Files changed (1) hide show

app.py CHANGED Viewed

@@ -125,7 +125,7 @@ def animate(
 @torch.no_grad()
 @spaces.GPU(duration=180)
 def animate_on_gpu(
-    image_data: Union[Image, List[Image]],
     seed: Optional[int] = 42,
     motion_bucket_id: int = 127,
     fps_id: int = 6,

 @torch.no_grad()
 @spaces.GPU(duration=180)
 def animate_on_gpu(
+    image_data: Union[PIL.Image.Image, List[PIL.Image.Image]],
     seed: Optional[int] = 42,
     motion_bucket_id: int = 127,
     fps_id: int = 6,