| { | |
| "guidance_scale_img": 4.5, | |
| "guidance_scale_text": 7.5, | |
| "config_p": "./configs/VITONHD_agnidmhand_agnv9hand_randdilate_gpt4ov6v12_jointcond_nopose_noip_intermcloth.yaml", | |
| "use_sd15": false, | |
| "resume_path": null, | |
| "pretrained_unet_path": null, | |
| "pretrained_unet_before_path": null, | |
| "gradient_accumulation_steps": 1, | |
| "prediction_type": "epsilon", | |
| "timestep_spacing": "leading", | |
| "noise_offset": 0.05, | |
| "num_inference_steps": 30, | |
| "finetuning": false, | |
| "save_step": 15000, | |
| "validation_step": 30000, | |
| "validation_step_lst": [ | |
| 2, | |
| 30000, | |
| 60000 | |
| ], | |
| "skip_paired": false, | |
| "skip_unpaired": false, | |
| "n_samples_train": 2, | |
| "n_samples_valid": 999999, | |
| "save_root_dir": "./logs", | |
| "save_name": "VITONHD_agnidmhand_agnv9hand_randdilate_gpt4ov6v12_jointcond_nopose_noip_intermcloth_bs32", | |
| "strength": 1.0, | |
| "img_h": 1024, | |
| "img_w": 768, | |
| "use_8bit_adam": false, | |
| "no_zero_snr": false, | |
| "init_model_path": "./pretrained_models/stable-diffusion-xl-1.0-inpainting-0.1", | |
| "init_vae_path": "./pretrained_models/sdxl-vae-fp16-fix", | |
| "init_ip_adapter_path": "./pretrained_models/IP-Adapter/sdxl_models/ip-adapter_sdxl.bin", | |
| "init_image_encoder_path": "./pretrained_models/IP-Adapter/sdxl_models", | |
| "init_cloth_encoder_path": "./pretrained_models/stable-diffusion-xl-base-1.0", | |
| "ip_adapter_num_tokens": 4, | |
| "logging_dir": "wandb", | |
| "learning_rate": 3.2e-05, | |
| "lr_scheduler": "constant_with_warmup", | |
| "no_lr_scheduler_accelerator": false, | |
| "lr_warmup_steps": 500, | |
| "lr_num_cycles": 1, | |
| "lr_power": 1.0, | |
| "weight_decay": 0.01, | |
| "guidance_scale": 2.0, | |
| "num_workers": 2, | |
| "mixed_precision": "fp16", | |
| "report_to": "wandb", | |
| "total_batch_size": 32 | |
| } |