参考:
https://huggingface.co/THUDM/CogVideoX-2b
4090卡24g是运行可能显存不足,这里暂时enable_model_cpu_offload()测试,并减少num_inference_steps步数
模型下载
export HF_ENDPOINT=https://hf-mirror.com
huggingface-cli download --resume-download --local-dir-use-symlinks False THUDM/CogVideoX-2b --local-dir cogvideox
包安装
pip install --upgrade opencv-python transformers
pip install