安装CogView4需要按照以下步骤进行环境配置:
硬件准备
建议使用配备NVIDIA GPU(至少16GB显存)的计算机,CPU也可运行但速度较慢
软件依赖
- 操作系统:支持Windows/Linux/macOS
- Python 3.8或更高版本
- PyTorch(推荐GPU版本,torch≥2.0)
- Git
安装步骤
- 克隆仓库:
git clone https://github.com/THUDM/CogView4.git
- 安装依赖:
pip install -r requirements.txt
- 下载预训练模型:可从Hugging Face手动下载或通过代码自动下载
- 配置优化选项:如
enable_model_cpu_offload()
等显存优化设置
注意:使用高分辨率(如2048×2048)需要24GB或更高显存的GPU,否则可能出现显存不足的情况。
本答案来源于文章《CogView4:生成中英双语高清图片的开源文生图模型》