CogVideo icon indicating copy to clipboard operation
CogVideo copied to clipboard

5b diffusers 版本生成视频时的显存占用大约4g,正常吗

Open dh12306 opened this issue 1 year ago • 1 comments

下载了5b的 diffusers版 ,文件大概40g, https://huggingface.co/THUDM/CogVideoX-5b 但是为什么加载之后生成的时候,显存只用了4g左右,是不是没加载对? QQ_1725956564364

启动用这个,python gradio_web_demo.py

dh12306 avatar Sep 10 '24 08:09 dh12306

跑的有点慢 QQ_1725957172644

dh12306 avatar Sep 10 '24 08:09 dh12306

这是正常的,但是很慢,因为已经开启了所有显存优化,如果你的GPU显存够,关闭cpu_load就行,cli_demo中有详细的解释。 readme中的数据是A100测的,如果你的GPU是T4,V100等,大概需要5-10分钟才能跑出一个视频

zRzRzRzRzRzRzR avatar Sep 11 '24 07:09 zRzRzRzRzRzRzR