Veldrovive commited on
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": 512,
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": 48,
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,