Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-02-24 12:10:09 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix lint issue

Browse Source
This commit is contained in:
mulin.lyh
2024-02-27 15:39:43 +08:00
parent 1a6eb34153
commit f7c4d5456c
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
modelscope/pipelines/multi_modal/text_to_video_synthesis_pipeline.py
View File

@@ -81,7 +81,12 @@ class TextToVideoSynthesisPipeline(Pipeline):
# Stack frames along a new dimension to create a 4D tensor (T, H, W, C)
imgs_tensor = torch.stack(frames, dim=0)
torchvision.io.write_video(output_video_path, imgs_tensor, fps=8, video_codec='h264', options={'crf': '10'})
torchvision.io.write_video(
output_video_path,
imgs_tensor,
fps=8,
video_codec='h264',
options={'crf': '10'})
if temp_video_file:
video_file_content = b''
with open(output_video_path, 'rb') as f: