Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
963d046
1
Parent(s):
5673a25
fix bugs
Browse files
app.py
CHANGED
@@ -94,7 +94,7 @@ def video_to_gif_and_images(video, indices):
|
|
94 |
]
|
95 |
|
96 |
|
97 |
-
def get_duration_single_image_to_long_video(idx: int, guidance_scale: float, fps: int):
|
98 |
return 30 * fps
|
99 |
|
100 |
|
@@ -121,7 +121,7 @@ def single_image_to_long_video(
|
|
121 |
return export_to_video(gen_video[0].detach().cpu(), fps=fps)
|
122 |
|
123 |
|
124 |
-
@spaces.GPU(duration=
|
125 |
@torch.autocast("cuda")
|
126 |
@torch.no_grad()
|
127 |
def any_images_to_short_video(
|
@@ -175,6 +175,15 @@ class CustomProgressBar:
|
|
175 |
def __getattr__(self, attr):
|
176 |
return getattr(self.pbar, attr)
|
177 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
178 |
|
179 |
@spaces.GPU(duration=45)
|
180 |
@torch.autocast("cuda")
|
@@ -365,7 +374,7 @@ def _interpolate_between(
|
|
365 |
|
366 |
|
367 |
def get_duration_smooth_navigation(
|
368 |
-
video: torch.Tensor, poses: torch.Tensor, interpolation_factor: int
|
369 |
):
|
370 |
length = (len(video) - 1) * interpolation_factor + 1
|
371 |
return 2 * length
|
|
|
94 |
]
|
95 |
|
96 |
|
97 |
+
def get_duration_single_image_to_long_video(idx: int, guidance_scale: float, fps: int, progress:gr.Progress):
|
98 |
return 30 * fps
|
99 |
|
100 |
|
|
|
121 |
return export_to_video(gen_video[0].detach().cpu(), fps=fps)
|
122 |
|
123 |
|
124 |
+
@spaces.GPU(duration=30)
|
125 |
@torch.autocast("cuda")
|
126 |
@torch.no_grad()
|
127 |
def any_images_to_short_video(
|
|
|
175 |
def __getattr__(self, attr):
|
176 |
return getattr(self.pbar, attr)
|
177 |
|
178 |
+
def get_duration_navigate_video(video: torch.Tensor,
|
179 |
+
poses: torch.Tensor,
|
180 |
+
x_angle: float,
|
181 |
+
y_angle: float,
|
182 |
+
distance: float
|
183 |
+
):
|
184 |
+
if abs(x_angle) < 30 and abs(y_angle) < 30 and distance < 150:
|
185 |
+
return 45
|
186 |
+
return 30
|
187 |
|
188 |
@spaces.GPU(duration=45)
|
189 |
@torch.autocast("cuda")
|
|
|
374 |
|
375 |
|
376 |
def get_duration_smooth_navigation(
|
377 |
+
video: torch.Tensor, poses: torch.Tensor, interpolation_factor: int, progress: gr.Progress
|
378 |
):
|
379 |
length = (len(video) - 1) * interpolation_factor + 1
|
380 |
return 2 * length
|