img2img-turbo-素描转图片,黑夜白天互转。。。。

源码地址 https://github.com/GaParmar/img2img-turbo

在线体验地址 https://huggingface.co/spaces/gparmar/img2img-turbo-sketch

安装

git clone https://github.com/GaParmar/img2img-turbo.git
cd img2img-turbo
conda env create -f environment.yaml

经过测试,triton这个库pip是装不上的,找不到,需要去https://huggingface.co/r4ziel/xformers_pre_built/tree/main 下载whl文件

或者你去其他地方先把文件下载下来

下载完成后,把environment.yaml和requirements.txt里面的triton==2.0.0删除掉,PS:应该早就删除的,只不过我踩了坑

手动安装,先激活上一步的环境 conda activate img2img-turbo

conda activate img2img-turbo
pip install triton-2.0.0-cp310-cp310-win_amd64.whl

最后再安装其他的

pip install -r requirements.txt

最后进行试验

python src/inference_paired.py --model_name "edge_to_image"   --input_image "assets/examples/bird.png"  --prompt "a blue bird"  --output_dir "outputs"

posted @ 2024-05-14 09:39  朝阳1  阅读(1)  评论(0编辑  收藏  举报