Append: set video compress quality as original version.

This commit is contained in:
Junming Chen
2025-12-14 20:53:26 +00:00
parent 127cc9007a
commit a4d34d9f3d
2 changed files with 2 additions and 2 deletions

View File

@@ -69,7 +69,7 @@ def speech_to_video(
motion_video = torch.cat((motion_video[:,:,overlap_frames_num:,:,:], current_clip_tensor[:,:,-overlap_frames_num:,:,:]), dim=2) motion_video = torch.cat((motion_video[:,:,overlap_frames_num:,:,:], current_clip_tensor[:,:,-overlap_frames_num:,:,:]), dim=2)
current_clip_quantized = pipe.vae_output_to_video(current_clip_tensor) current_clip_quantized = pipe.vae_output_to_video(current_clip_tensor)
video.extend(current_clip_quantized) video.extend(current_clip_quantized)
save_video_with_audio(video, save_path, audio_path, fps=16, quality=10) save_video_with_audio(video, save_path, audio_path, fps=16, quality=5)
print(f"processed the {r+1}th clip of total {num_repeat} clips.") print(f"processed the {r+1}th clip of total {num_repeat} clips.")
return video return video

View File

@@ -68,7 +68,7 @@ def speech_to_video(
motion_video = torch.cat((motion_video[:,:,overlap_frames_num:,:,:], current_clip_tensor[:,:,-overlap_frames_num:,:,:]), dim=2) motion_video = torch.cat((motion_video[:,:,overlap_frames_num:,:,:], current_clip_tensor[:,:,-overlap_frames_num:,:,:]), dim=2)
current_clip_quantized = pipe.vae_output_to_video(current_clip_tensor) current_clip_quantized = pipe.vae_output_to_video(current_clip_tensor)
video.extend(current_clip_quantized) video.extend(current_clip_quantized)
save_video_with_audio(video, save_path, audio_path, fps=16, quality=10) save_video_with_audio(video, save_path, audio_path, fps=16, quality=5)
print(f"processed the {r+1}th clip of total {num_repeat} clips.") print(f"processed the {r+1}th clip of total {num_repeat} clips.")
return video return video