昇腾社区首页
中文
注册

def __call__

函数功能

OpenSoraPipeline文生视频推理接口。

函数原型

def __call__(self, prompts, seed=42, output_type="latent"): -> Torch.Tensor

参数说明

参数名

输入/输出

说明

prompts

输入

用户输入文本,比如["a sunny day, cloud moving slowly over a lush forest"]。

seed

输入

随机种子,默认值为42。

output_type

输入

输出的类型。

  • 若该值为“latent”,则不对结果进行后处理;
  • 若该值为“thwc”,则对视频数据进行后处理,处理成thwc的四维tensor,后续可直接调用torchvision.io.write_video(save_path, video, fps=8, video_codec="h264")保存视频。

返回值说明

视频生成结果。