Upload config.json with huggingface_hub

config.json

@@ -1,21 +1,23 @@
 {
   "output_dir": "./outputs",
   "model_repo": "AbstractPhil/sd15-flow-lune",
-  "checkpoint_filename": "sd15_flow_lune_flux_e9_s12500.pt",
-  "dataset_name": "irodkin/ffhq_with_llava_shorter_captions",
+  "checkpoint_filename": "sd15_flow_lune_e52_s52296.pt",
+  "dataset_name": "AbstractPhil/CN_pose3D_V10_512",
   "hf_repo_id": "AbstractPhil/sd15-flow-lune",
   "upload_to_hub": true,
+  "run_name": "pretrain_pose_controlnet_t500_700",
   "continue_from_checkpoint": false,
   "seed": 42,
   "batch_size": 16,
   "base_lr": 2e-06,
-  "shift": 3.0,
+  "shift": 2.5,
   "dropout": 0.1,
+  "min_snr_gamma": 5.0,
   "min_timestep": 500.0,
-  "max_timestep": 1000.0,
-  "num_train_epochs": 1,
+  "max_timestep": 700.0,
+  "num_train_epochs": 2,
   "warmup_epochs": 1,
-  "checkpointing_steps": 1000,
+  "checkpointing_steps": 2500,
   "num_workers": 0,
   "vae_scale": 0.18215
 }