Upload config.json with huggingface_hub

config.json

@@ -1,17 +1,21 @@
 {
   "output_dir": "./outputs",
   "model_repo": "AbstractPhil/sd15-flow-lune",
-  "checkpoint_filename": "sd15_flow_lune_e34_s34000.pt",
-  "dataset_name": "AbstractPhil/sd15-latent-distillation-500k",
+  "checkpoint_filename": "sd15_flow_lune_flux_e9_s12500.pt",
+  "dataset_name": "irodkin/ffhq_with_llava_shorter_captions",
   "hf_repo_id": "AbstractPhil/sd15-flow-lune",
   "upload_to_hub": true,
+  "continue_from_checkpoint": false,
   "seed": 42,
   "batch_size": 16,
   "base_lr": 2e-06,
-  "shift": 2.5,
+  "shift": 3.0,
   "dropout": 0.1,
-  "max_train_steps": 100000,
-  "checkpointing_steps": 2000,
+  "min_timestep": 500.0,
+  "max_timestep": 1000.0,
+  "num_train_epochs": 1,
+  "warmup_epochs": 1,
+  "checkpointing_steps": 1000,
   "num_workers": 0,
   "vae_scale": 0.18215
 }