Veldrovive
commited on
Commit
•
a450088
1
Parent(s):
de501e0
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 |
"image_embed_dim": 768,
|
9 |
"text_embed_dim": 768,
|
10 |
"cond_on_text_encodings": false,
|
@@ -29,7 +29,7 @@
|
|
29 |
"webdataset_base_url": "pipe:aws s3 cp --quiet s3://s-datasets/laion-aesthetic/data/laion2B-en-aesthetic/{}.tar -",
|
30 |
"img_embeddings_url": "s3://s-datasets/laion-aesthetic/ordered_embeddings/",
|
31 |
"num_workers": 6,
|
32 |
-
"batch_size":
|
33 |
"start_shard": 0,
|
34 |
"end_shard": 5247,
|
35 |
"shard_width": 5,
|
|
|
4 |
"decoder": {
|
5 |
"unets": [
|
6 |
{
|
7 |
+
"dim": 576,
|
8 |
"image_embed_dim": 768,
|
9 |
"text_embed_dim": 768,
|
10 |
"cond_on_text_encodings": false,
|
|
|
29 |
"webdataset_base_url": "pipe:aws s3 cp --quiet s3://s-datasets/laion-aesthetic/data/laion2B-en-aesthetic/{}.tar -",
|
30 |
"img_embeddings_url": "s3://s-datasets/laion-aesthetic/ordered_embeddings/",
|
31 |
"num_workers": 6,
|
32 |
+
"batch_size": 40,
|
33 |
"start_shard": 0,
|
34 |
"end_shard": 5247,
|
35 |
"shard_width": 5,
|