环境安装
根据Kolors【github】的指引,安装命令如下:
apt-get install git-lfs
git clone https://github.com/Kwai-Kolors/Kolors
cd Kolors
conda create --name kolors python=3.8
conda activate kolors
pip install -r requirements.txt
python3 setup.py install
但是不知道为什么,requirements里面的内容有冲突,最后修改如下
fire
triton
pydantic==2.8.2
accelerate==0.27.2
deepspeed==0.8.1
huggingface-hub==0.23.4
imageio==2.25.1
numpy==1.21.6
omegaconf==2.3.0
pandas==1.3.5
Pillow==9.4.0
tokenizers==0.19.0
torch==1.13.1
torchvision==0.14.1
transformers==4.42.4
xformers==0.0.16
safetensors
diffusers==0.28.2
sentencepiece==0.1.99
gradio==4.37.2
opencv-python
einops
timm
onnxruntime
有一个技巧是,你可以跳过上面的步骤,安几个包后直接用腾讯混元的环境来运行项目。
模型下载
一开始使用git lfs clone https://huggingface.co/Kwai-Kolors/Kolors weights/Kolors
下载,但是文件校验出错,模型推理也会报错。后来把text_encoder文件夹里面的内容都删了,再执行export HF_ENDPOINT=https://hf-mirror.com huggingface-cli download --resume-download Kwai-Kolors/Kolors --local-dir weights/Kolors
这两句才成功。
页面运行
执行python3 scripts/sampleui.py
,结果如下