| { | |
| "output_dir": "./outputs", | |
| "model_repo": "AbstractPhil/sd15-flow-lune", | |
| "checkpoint_filename": "sd15_flow_lune_e19_s19000.pt", | |
| "dataset_name": "AbstractPhil/sd15-latent-distillation-500k", | |
| "hf_repo_id": "AbstractPhil/sd15-flow-lune", | |
| "upload_to_hub": true, | |
| "seed": 42, | |
| "batch_size": 16, | |
| "base_lr": 2e-06, | |
| "shift": 2.0, | |
| "dropout": 0.1, | |
| "max_train_steps": 50000, | |
| "checkpointing_steps": 2000, | |
| "num_workers": 0, | |
| "vae_scale": 0.18215 | |
| } |