文章目录
- 1 图像生成模型 Kolors
- 2 部署流程
- 3 部署服务
- 3.1 安装 Anaconda
- 3.2 Kolors 库下载
- 3.3 创建虚拟环境
- 4 生成图片
1 图像生成模型 Kolors
Kolors是由快手团队开发的大规模文本到图像生成模型,以其独特的潜在扩散技术而闻名。
Kolors通过在数十亿对文本和图像上进行训练,在视觉质量、语义理解及中英文文本渲染方面超越许多开源和专有模型,并且在处理中文文本时展现了强大的理解和生成能力。
Kolors的研发基于Stable Diffusion框架,这一框架已经在文本转图像生成方面表现出色。通过整合潜在扩散技术,Kolors不仅提升了图像质量,还能更细致地捕捉文本的深层含义和意图。
要使用Kolors进行文本到图像转换,用户需准备好特定的软件环境,包括Python 3.8及以上版本、PyTorch 1.13.1及以上版本,以及Transformers 4.26.1及以上版本。
建议使用CUDA 11.7或更高版本以加速模型性能,用户可通过Anaconda来管理Python环境,并创建独立的虚拟环境以避免依赖冲突。准备好环境后,用户可以下载Kolors库及其依赖项,并通过命令行执行图像生成。
前提条件
- Python 3.8或更高版本
- PyTorch 1.13.1或更新版本
- Transformers 4.26.1或更新版本
- 推荐使用CUDA 11.7或更高版本
2 部署流程
进入丹摩DAMODEL|让AI开发更简单!算力租赁上丹摩!
(1)创建实例。
(2)进入创建页面后,首先在实例配置中选择付费类型,一般短期需求可以选择按量付费或者包日,长期需求可以选择包月套餐。
(3)选择GPU数量和需求的GPU型号,首次创建实例推荐选择:
- 按量付费–GPU数量1–NVIDIA-GeForc-RTX-4090,该配置为60GB内存,24GB的显存。
- 配置数据硬盘的大小,每个实例默认附带了50GB的数据硬盘,首次创建可以就选择默认大小50GB。
(4)继续选择安装的镜像,平台提供了一些基础镜像供快速启动,镜像中安装了对应的基础环境和框架,可通过勾选来筛选框架,这里筛选PyTorch,选择PyTorch 2.4.0。
(5)为保证安全登录,创建密钥对,输入自定义的名称,然后选择自动创建并将创建好的私钥保存的自己电脑中并将后缀改为.pem,以便后续本地连接使用。
(6)创建好密钥对后,选择刚刚创建好的密钥对,并点击立即创建,等待一段时间后即可启动成功!
3 部署服务
3.1 安装 Anaconda
(1)下载安装脚本
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
(2)安装 Anaconda*
bash Miniconda3-latest-Linux-x86_64.sh
(3)加入环境变量
vim ~/.bashrc
export PATH=$PATH:/root/miniconda3/condabin
(4)重新加载环境变量
source ~/.bashrc
3.2 Kolors 库下载
(1)安装Git LFS
apt-get install git-lfs
(2)克隆项目
git clone https://github.com/Kwai-Kolors/Kolors
cd Kolors
3.3 创建虚拟环境
(1)安装Python
conda create --name kolors python=3.8
(2)初始化bash
conda init bash
(3)激活虚拟环境
source ~/.bashrc
conda activate kolors
(4)安装依赖项
pip install -r requirements.txt
(5)安装 Kolors
python3 setup.py install
(6)克隆仓库
git lfs clone https://huggingface.co/Kwai-Kolors/Kolors weights/Kolors
4 生成图片
当服务部署完成后,就可以使用 Kolors 生成图片了。运行以下命令:
python3 scripts/sample.py "一个在星空下奔跑的女孩"
# The image will be saved to "scripts/outputs/sample_test.jpg"
生成的图像会被保存为scripts/outputs/sample_test.jpg
。
可以根据自己的想法修改文本描述,以生成各种风格独特的图像。通过这些简单的步骤,您就能成功部署并使用Kolors文本到图像生成工具。
利用Kolors,可以将脑海中的创意转化为视觉艺术,创造出令人耳目一新的图像作品。无论是独特的艺术作品还是生动的场景,Kolors都能帮助您实现愿景,快来掌握这项新技术,并用它来打造属于自己的图像作品吧!