This project is a Industrial Defect Diffusion Generater which named as IDDG.
1. git clone https://gitlab.svfactory.com/Sunnyway/iddg
2. cd ./iddg
3. pip install -r requirements.txtLoRA works as a plugins in IDDG. So, to train a LoRA model, you need download the Diffusion Model [NAS PATH: /homes/sunhuanrong/maliang/ldm/v2.0.0].
1. cd ./iddg
2. python scripts/img_crop/crop_rdir_imgs_for_lora.pyconfig/train_lora/train_lora_inpainting_ldm_256.json
1. python run.py -p train -c config/train_lora/train_lora_inpainting_ldm_256.json