Upload decoder/decoder_config.json with huggingface_hub

decoder/decoder_config.json

@@ -3,13 +3,13 @@
     "decoder": {
         "unets": [
             {
-                "dim": 384,
+                "dim": 320,
                 "cond_dim": 512,
                 "image_embed_dim": 1024,
                 "text_embed_dim": 1024,
                 "cond_on_text_encodings": true,
                 "channels": 3,
-                "dim_mults": [1, 2, 2, 4],
+                "dim_mults": [1, 2, 3, 4],
                 "num_resnet_blocks": 4,
                 "attn_heads": 8,
                 "attn_dim_head": 64,
@@ -33,9 +33,8 @@
     },
     "data": {
         "webdataset_base_url": "pipe:aws s3 cp --quiet s3://s-datasets/laion5b/laion2B-data/{}.tar -", 
-        "img_embeddings_url": "s3://s-laion/vit-h-14-embeddings/img_emb/",
         "num_workers": 6,
-        "batch_size": 1,
+        "batch_size": 16,
         "start_shard": 0,
         "end_shard": 231349,
         "shard_width": 6,