Upload decoder_config.json with huggingface_hub

decoder_config.json

@@ -4,7 +4,7 @@
     "decoder": {
         "unets": [
             {
-                "dim": 256,
+                "dim": 64,
                 "image_embed_dim": 768,
                 "channels": 3,
                 "dim_mults": [1, 2, 3, 4]
@@ -47,11 +47,11 @@
         "lr":5e-5,
         "wd": 0.1,
         "max_grad_norm": 0.5,
-        "save_every_n_samples": 1000,
+        "save_every_n_samples": 300,
         "n_sample_images": 10,
         "device": "cuda:0",
-        "epoch_samples": 1500,
-        "validation_samples": 1000,
+        "epoch_samples": 500,
+        "validation_samples": 500,
         "use_ema": true,
         "ema_beta": 0.99,
         "unet_training_mask": [true]