Upload folder using huggingface_hub

config.json

@@ -14,12 +14,15 @@
   "beta_start": 0.0001,
   "beta_end": 0.02,
   "learning_rate": 0.0001,
-  "vae_pretrain_epochs": 1,
+  "vae_pretrain_epochs": 10,
   "vae_pretrain_lr": 0.001,
+  "kl_weight": 0.1,
+  "cfg_dropout_prob": 0.15,
   "grad_clip_norm": 1.0,
   "warmup_steps": 500,
   "batch_size": 8,
   "num_epochs": 30,
+  "dataset_buffer_size": 2000,
   "dropout": 0.1,
   "weight_decay": 0.01,
   "early_stop_patience": 8,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d3427674b4880b0669b14197deeb45e6dbdbafad55290f32c35b6cd1bcebd477
-size 267847332
+oid sha256:ad2f679f3bc1378b4cdc9aec06b2c5f548595049bb69e6f0cdb6dd007caba44f
+size 267848956