Upload config.json with huggingface_hub
Browse files- config.json +16 -0
config.json
ADDED
|
@@ -0,0 +1,16 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"output_dir": "./outputs",
|
| 3 |
+
"model_path": "AbstractPhil/sd15-flow-matching-try2",
|
| 4 |
+
"checkpoint_file": "sd15_flowmatch_david_weighted_2_e34.pt",
|
| 5 |
+
"dataset_name": "AbstractPhil/sd15-latent-distillation-500k",
|
| 6 |
+
"hf_repo_id": "AbstractPhil/sd15-flow-lune",
|
| 7 |
+
"upload_to_hub": true,
|
| 8 |
+
"seed": 42,
|
| 9 |
+
"batch_size": 16,
|
| 10 |
+
"base_lr": 2e-06,
|
| 11 |
+
"shift": 2.0,
|
| 12 |
+
"dropout": 0.1,
|
| 13 |
+
"max_train_steps": 50000,
|
| 14 |
+
"checkpointing_steps": 1000,
|
| 15 |
+
"num_workers": 4
|
| 16 |
+
}
|