Spaces:
Runtime error
Runtime error
fix-pipelines
#4
by naicoi - opened
.github/workflows/docker.yml
CHANGED
|
@@ -14,7 +14,7 @@ env:
|
|
| 14 |
|
| 15 |
jobs:
|
| 16 |
build:
|
| 17 |
-
runs-on:
|
| 18 |
permissions:
|
| 19 |
contents: read
|
| 20 |
packages: write
|
|
|
|
| 14 |
|
| 15 |
jobs:
|
| 16 |
build:
|
| 17 |
+
runs-on: self-hosted
|
| 18 |
permissions:
|
| 19 |
contents: read
|
| 20 |
packages: write
|
latentsync/pipelines/lipsync_pipeline.py
CHANGED
|
@@ -606,8 +606,10 @@ class LipsyncPipeline(DiffusionPipeline):
|
|
| 606 |
torch.cuda.empty_cache()
|
| 607 |
|
| 608 |
# Restore video from CPU tensors to save GPU memory
|
|
|
|
|
|
|
| 609 |
synced_video_frames = self.restore_video_from_cpu(
|
| 610 |
-
|
| 611 |
)
|
| 612 |
|
| 613 |
audio_samples_remain_length = int(
|
|
|
|
| 606 |
torch.cuda.empty_cache()
|
| 607 |
|
| 608 |
# Restore video from CPU tensors to save GPU memory
|
| 609 |
+
# Flatten synced_video_frames: list of (B, C, H, W) -> list of (C, H, W)
|
| 610 |
+
flattened_frames = [frame for batch in synced_video_frames for frame in batch]
|
| 611 |
synced_video_frames = self.restore_video_from_cpu(
|
| 612 |
+
flattened_frames, video_frames, boxes, affine_matrices
|
| 613 |
)
|
| 614 |
|
| 615 |
audio_samples_remain_length = int(
|