PaddleMIX
PaddleMIX copied to clipboard
Custom Diffusion 训练支持
Custom Diffusion 训练支持
任务描述
任务背景
- PaddleMIX ppdiffusers新增custom_diffusion训练流程。
完成步骤
- 参考代码完成对齐
提交内容:
- 提交到目录
https://github.com/huggingface/diffusers/tree/main/examples/custom_diffusion
pip install clip-retrieval
python retrieve.py --class_prompt cat --class_data_dir real_reg/samples_cat --num_class_images 200
运行 retrieve.py 获取数据集有错误
修改clip-retrieval调试返回信息是服务器会返回502错误