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": 640,
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": 32,
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,