IDM-VTON效果测试
地址:https://github.com/yisol/IDM-VTON
模型和配置准备:
环境配置:
registry.cn-beijing.aliyuncs.com/modelscope-repo/modelscope:ubuntu22.04-cuda12.1.0-py310-torch2.1.2-tf2.14.0-1.13.1
报错:ImportError: cannot import name ‘PositionNet‘ from ‘diffusers.models.embeddings‘
修改:https://blog.csdn.net/qq_34342853/article/details/140459724
报错:OSError: yisol/IDM-VTON does not appear to have a file named scheduler_config.json.
修改:设置环境变量HF_ENDPOINT为https://hf-mirror.com即可。
推理:
python inference.py --width 768 --height 1024 --num_inference_steps 30 --output_dir “result” --unpaired --data_dir “DATA_DIR” --seed 42 --test_batch_size 2 --guidance_scale 2.0