Upload config.json with huggingface_hub
Browse files- config.json +3 -3
config.json
CHANGED
|
@@ -1,14 +1,14 @@
|
|
| 1 |
{
|
| 2 |
"output_dir": "./outputs",
|
| 3 |
-
"
|
| 4 |
-
"
|
| 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":
|
| 12 |
"dropout": 0.1,
|
| 13 |
"max_train_steps": 50000,
|
| 14 |
"checkpointing_steps": 1000,
|
|
|
|
| 1 |
{
|
| 2 |
"output_dir": "./outputs",
|
| 3 |
+
"model_repo": "AbstractPhil/sd15-flow-matching-try2",
|
| 4 |
+
"checkpoint_filename": "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,
|