安装CogView4需要按照以下步骤进行环境配置:
ハードウェアの準備
建议使用配备NVIDIA GPU(至少16GB显存)的计算机,CPU也可运行但速度较慢
ソフトウェア依存
- 操作系统:支持Windows/Linux/macOS
- Python 3.8以上
- PyTorch(推荐GPU版本,torch≥2.0)
- ギット
インストール手順
- クローン倉庫
git clone https://github.com/THUDM/CogView4.git
- 依存関係をインストールします:
pip install -r requirements.txt
- 下载预训练模型:可从Hugging Face手动下载或通过代码自动下载
- 配置优化选项:如
enable_model_cpu_offload()
等显存优化设置
注意:使用高分辨率(如2048×2048)需要24GB或更高显存的GPU,否则可能出现显存不足的情况。
この答えは記事から得たものである。CogView4:中国語と英語の高精細対訳画像を生成するオープンソースのグラフィカルモデルについて