Spaces:
Running
on
Zero
Running
on
Zero
added gpu declaration
Browse files
app.py
CHANGED
@@ -98,6 +98,7 @@ def get_colors(num_colors: int) -> List[Tuple[int, int, int]]:
|
|
98 |
# )
|
99 |
# return torch.stack([grid_x, grid_y], dim=-1).reshape(1, -1, 2)
|
100 |
|
|
|
101 |
def paint_point_track_gpu_scatter(
|
102 |
frames: np.ndarray,
|
103 |
point_tracks: np.ndarray,
|
|
|
98 |
# )
|
99 |
# return torch.stack([grid_x, grid_y], dim=-1).reshape(1, -1, 2)
|
100 |
|
101 |
+
@spaces.GPU
|
102 |
def paint_point_track_gpu_scatter(
|
103 |
frames: np.ndarray,
|
104 |
point_tracks: np.ndarray,
|