PhotoMaker 人像风格迁移
不用训练人物 lora,直接用一张图片生成不同风格的照片
1 安装
conda create --name photomaker python=3.10 -y
conda activate photomaker
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 cudnn=8.1.0 -c pytorch -c nvidia -y
pip install -U pip
git clone https://github.com/TencentARC/PhotoMaker
cd PhotoMaker
pip install -r requirements.txt
pip install git+https://github.com/TencentARC/PhotoMaker.git
pip install einops onnxruntime-gpu
编辑 /root/miniconda3/envs/photomaker/lib/python3.10/site-packages/photomaker/pipeline_t2i_adapter.py
注释掉 resume_download=resume_download,,如下:
if not isinstance(pretrained_model_name_or_path_or_dict, dict):
model_file = _get_model_file(
pretrained_model_name_or_path_or_dict,
weights_name=weight_name,
cache_dir=cache_dir,
force_download=force_download,
# resume_download=resume_download,
proxies=proxies,
local_files_only=local_files_only,
token=token,
revision=revision,
subfolder=subfolder,
user_agent=user_agent,
)
修改 gradio_demo/app_v2.py,最后一行改为 demo.launch(server_name="0.0.0.0")
2 运行
#!/bin/bash
export CUDA_VISIBLE_DEVICES=0
source /root/miniconda3/bin/activate photomaker
cd /home/PhotoMaker
python gradio_demo/app_v2.py