laion
/

DALLE2-PyTorch

rom1504 commited on Aug 15, 2022

Commit

a95b7ea

1 Parent(s): 160894b

Upload decoder/laion_august/decoder_config.json with huggingface_hub

Files changed (1) hide show

decoder/laion_august/decoder_config.json CHANGED Viewed

@@ -3,7 +3,7 @@
     "decoder": {
         "unets": [
             {
-                "dim": 320,
                 "cond_dim": 512,
                 "image_embed_dim": 768,
                 "text_embed_dim": 768,
@@ -34,7 +34,7 @@
     "data": {
         "webdataset_base_url": "pipe:aws s3 cp --quiet s3://s-datasets/laion5b/laion2B-data/{}.tar -",
         "num_workers": 12,
-        "batch_size": 70,
         "start_shard": 0,
         "end_shard": 231349,
         "shard_width": 6,

     "decoder": {
         "unets": [
             {
+                "dim": 576,
                 "cond_dim": 512,
                 "image_embed_dim": 768,
                 "text_embed_dim": 768,
     "data": {
         "webdataset_base_url": "pipe:aws s3 cp --quiet s3://s-datasets/laion5b/laion2B-data/{}.tar -",
         "num_workers": 12,
+        "batch_size": 40,
         "start_shard": 0,
         "end_shard": 231349,
         "shard_width": 6,