昇腾社区首页
中文
注册

def __call__

函数功能

OpenSoraPipeline文生视频推理接口。

函数原型

def __call__(self, prompts: List[str], seed: int = 42, output_type: str = "latent"): -> torch.Tensor

参数说明

参数名

输入/输出

类型

说明

prompts

输入

List[str]

用户输入文本,如:["a sunny day, cloud moving slowly over a lush forest"],需要由用户保证输入prompts的合法性。

List长度目前只支持1,输入的提示词长度要求在(0,1024)之间。

seed

输入

int

生成随机数据种子;默认值为42。

output_type

输入

str

输出的类型。

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

返回值说明

视频生成结果。