一、部署说明

此模型需使用flash_attn进行加速推理,对老GPU的支持不佳,导致很多gpu无法安装词库或无法推理(报错:RuntimeError: FlashAttention only supports Ampere GPUs or newer.)。有大佬提供了使用xformers进行替代,才得以运行,详情可去github上的讨论区查看。

二、测试硬件

RTX8000

三、测试样例

注:视频已转换为0.3倍尺寸,帧率调整为10的GIF图片

3.1 样例1

荷花已经从水里长出来开花了

python generate.py  --task t2v-1.3B --size 832*480 --ckpt_dir /Wan2.1/Wan-AI-Wan2.1-T2V-1.3B --sample_shift 8 --sample_guide_scale 6 --prompt "The lotus has been growing out of the water and flowering"

耗时:约30min
显存消耗:18.5G
在这里插入图片描述
内存消耗:>21.6G
视频:

t2v-1.3B_832-480荷花

3.2 样例2

两只拟人化的猫穿着舒适的拳击装备,戴着明亮的手套,在聚光灯下的舞台上激烈地搏斗。

python generate.py --task t2v-1.3B --size 832*480 --ckpt_dir /Wan2.1/Wan-AI-Wan2.1-T2V-1.3B --sample_shift 8 --sample_guide_scale 6 --prompt "Two anthropomorphic cats in comfy boxing gear and bright gloves fight intensely on a spotlighted stage."

t2v-1.3B_832-480拳击猫

后续测试抽空继续…

Logo

技术共进,成长同行——讯飞AI开发者社区

更多推荐