CogVideo
CogVideo copied to clipboard
5b diffusers 版本生成视频时的显存占用大约4g,正常吗
下载了5b的 diffusers版 ,文件大概40g,
https://huggingface.co/THUDM/CogVideoX-5b
但是为什么加载之后生成的时候,显存只用了4g左右,是不是没加载对?
启动用这个,python gradio_web_demo.py
跑的有点慢
这是正常的,但是很慢,因为已经开启了所有显存优化,如果你的GPU显存够,关闭cpu_load就行,cli_demo中有详细的解释。 readme中的数据是A100测的,如果你的GPU是T4,V100等,大概需要5-10分钟才能跑出一个视频