| { | |
| "output_dir": "./outputs", | |
| "model_repo": "AbstractPhil/sd15-flow-lune", | |
| "checkpoint_filename": "sd15_flow_pretrain_pose_controlnet_t500_700_s8312.pt", | |
| "dataset_name": "AbstractPhil/CN_pose3D_V7_512", | |
| "use_masks": true, | |
| "mask_column": "mask", | |
| "hf_repo_id": "AbstractPhil/sd15-flow-lune", | |
| "upload_to_hub": true, | |
| "run_name": "pretrain_pose_controlnet_v7_v10_t400_600", | |
| "continue_from_checkpoint": false, | |
| "seed": 42, | |
| "batch_size": 32, | |
| "base_lr": 2e-06, | |
| "shift": 2.5, | |
| "dropout": 0.1, | |
| "min_snr_gamma": 5.0, | |
| "min_timestep": 400.0, | |
| "max_timestep": 600.0, | |
| "num_train_epochs": 1, | |
| "warmup_epochs": 1, | |
| "checkpointing_steps": 2500, | |
| "num_workers": 0, | |
| "vae_scale": 0.18215, | |
| "delimiter": ",", | |
| "preserved_count": 2, | |
| "remove_these": [ | |
| "simple background", | |
| "white background" | |
| ], | |
| "prepend_prompt": "doll", | |
| "append_prompt": "transparent background", | |
| "shuffle_prompt": true | |
| } |