def forward
函数功能
生成视频帧。
函数原型
def forward(self, x, timestep, y, mask=None, x_mask=None, fps=None, height=None, width=None, t_idx=0, **kwargs)
参数说明
参数名 |
输入/输出 |
说明 |
|---|---|---|
x |
输入 |
噪声张量，当前仅支持5维输入。 |
timestep |
输入 |
时间步张量，当前仅支持1维输入。 |
y |
输入 |
文本编码后的prompts输入，当前仅支持4维输入。 |
mask |
输入 |
text文本掩码，当前仅支持2维输入。 可选输入，默认为None。 |
x_mask |
输入 |
空间维度掩码，当前仅支持2维输入。 可选输入，默认为None。 |
fps |
输入 |
每秒生成视频的帧数，要求1维输入(B)。当输入为None时，代表每秒生成视频的帧数为8。 可选输入，默认为None。 |
height |
输入 |
生成视频的高，要求1维输入(B)。当输入为None时，代表生成视频的高为720。 可选输入，默认为None。 |
width |
输入 |
生成视频的宽，要求1维输入(B)。当输入为None时，代表生成视频的宽为1280。 可选输入，默认为None。 |
t_idx |
输入 |
当前迭代步索引，当配置use_cache为True时必须输入，用户需要保证输入的值小于最大迭代步数。 可选输入，默认值为0。 |
kwargs |
输入 |
其他参数。 |
返回值说明
返回生成的视频帧。