安装CogView4需要按照以下步骤进行环境配置:
Hardware preparation
建议使用配备NVIDIA GPU(至少16GB显存)的计算机,CPU也可运行但速度较慢
software dependency
- 操作系统:支持Windows/Linux/macOS
- Python 3.8 or higher
- PyTorch(推荐GPU版本,torch≥2.0)
- Git
Installation steps
- Cloning Warehouse:
git clone https://github.com/THUDM/CogView4.git
- Install the dependencies:
pip install -r requirements.txt
- 下载预训练模型:可从Hugging Face手动下载或通过代码自动下载
- 配置优化选项:如
enable_model_cpu_offload()
等显存优化设置
注意:使用高分辨率(如2048×2048)需要24GB或更高显存的GPU,否则可能出现显存不足的情况。
This answer comes from the articleCogView4: An Open Source Literature Graph Model for Generating Bilingual HD ImagesThe