.github/workflows/docker.yml CHANGED
@@ -14,7 +14,7 @@ env:
14
 
15
  jobs:
16
  build:
17
- runs-on: ubuntu-latest
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
- synced_video_frames, video_frames, boxes, affine_matrices
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(