Tencent / HunyuanVideo

HunyuanVideo: A Systematic Framework For Large Video Generation Model Training
https://aivideo.hunyuan.tencent.com
Other
2.64k stars 142 forks source link

batch infer 问题 #43

Open ChinChyi opened 13 hours ago

ChinChyi commented 13 hours ago

官方脚本中的推理保存video,有一点小问题, `

Save samples

for i, sample in enumerate(samples):
    sample = samples[i].unsqueeze(0)
    time_flag = datetime.fromtimestamp(time.time()).strftime("%Y-%m-%d-%H:%M:%S")
    save_path = f"{save_path}/{time_flag}_seed{outputs['seeds'][i]}_{outputs['prompts'][i][:100].replace('/','')}.mp4"
    save_videos_grid(sample, save_path, fps=24)
    logger.info(f'Sample save to: {save_path}')

` 如果是批量生成,这个save_path会无限递增,应该是args.save_path

JacobKong commented 12 hours ago

感谢您的反馈,我们会尽快修复这个问题