Commit
·
43e8866
1
Parent(s):
92b32b4
Upload decoder/2.4B_fp16/decoder_config.json with huggingface_hub
Browse files
decoder/2.4B_fp16/decoder_config.json
CHANGED
@@ -4,7 +4,7 @@
|
|
4 |
"decoder": {
|
5 |
"unets": [
|
6 |
{
|
7 |
-
"dim":
|
8 |
"cond_dim": 512,
|
9 |
"image_embed_dim": 768,
|
10 |
"text_embed_dim": 768,
|
@@ -30,7 +30,7 @@
|
|
30 |
"webdataset_base_url": "pipe:aws s3 cp --quiet s3://s-datasets/laion-aesthetic/data/laion2B-en-aesthetic/{}.tar -",
|
31 |
"img_embeddings_url": "s3://s-datasets/laion-aesthetic/ordered_embeddings/",
|
32 |
"num_workers": 6,
|
33 |
-
"batch_size":
|
34 |
"start_shard": 0,
|
35 |
"end_shard": 5247,
|
36 |
"shard_width": 5,
|
|
|
4 |
"decoder": {
|
5 |
"unets": [
|
6 |
{
|
7 |
+
"dim": 768,
|
8 |
"cond_dim": 512,
|
9 |
"image_embed_dim": 768,
|
10 |
"text_embed_dim": 768,
|
|
|
30 |
"webdataset_base_url": "pipe:aws s3 cp --quiet s3://s-datasets/laion-aesthetic/data/laion2B-en-aesthetic/{}.tar -",
|
31 |
"img_embeddings_url": "s3://s-datasets/laion-aesthetic/ordered_embeddings/",
|
32 |
"num_workers": 6,
|
33 |
+
"batch_size": 32,
|
34 |
"start_shard": 0,
|
35 |
"end_shard": 5247,
|
36 |
"shard_width": 5,
|