Upload folder using huggingface_hub

config.json

@@ -14,12 +14,16 @@
   "beta_start": 0.0001,
   "beta_end": 0.02,
   "learning_rate": 0.0001,
-  "vae_pretrain_epochs": 1,
+  "vae_pretrain_epochs": 2,
   "vae_pretrain_lr": 0.001,
   "grad_clip_norm": 1.0,
   "warmup_steps": 500,
   "batch_size": 8,
-  "num_epochs": 5,
+  "num_epochs": 30,
+  "dropout": 0.1,
+  "weight_decay": 0.01,
+  "early_stop_patience": 8,
+  "ema_decay": 0.999,
   "model_name": "ShellD",
   "_class_name": "ShellD",
   "_version": "0.1"

model.safetensors

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