pipe.vae.enable_slicing() 用了为什么现场还需要50多G - Githubissues

THUDM / CogVideo

text and image to video generation: CogVideoX (2024) and CogVideo (ICLR 2023)

Apache License 2.0

8.4k stars 803 forks source link

pipe.vae.enable_slicing() 用了为什么现场还需要50多G #232

Closed lonngxiang closed 1 month ago

lonngxiang commented 2 months ago

System Info / 系統信息

https://colab.research.google.com/drive/14TTaDTM3_lk69qKb5u4-1_gm_YK6lM3m?usp=sharing 代码版本

Information / 问题信息

[ ] The official example scripts / 官方的示例脚本
[ ] My own modified scripts / 我自己修改的脚本和任务

Reproduction / 复现过程

https://colab.research.google.com/drive/14TTaDTM3_lk69qKb5u4-1_gm_YK6lM3m?usp=sharing 代码

Expected behavior / 期待表现

正常运行

zRzRzRzRzRzRzR commented 2 months ago

请按照issue置顶的方案解决