Hugging Face's logo

gsurma
/
ai_dreamer

Text-to-Image
JAX
English
generic
bart
Model card Files Community
1
gsurma commited on
Commit
f2ce52b
•
1 Parent(s): 4ccf723

no message

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -22,7 +22,7 @@ from vqgan_jax.modeling_flax_vqgan import VQModel
22
 
23
  # Model hyperparameters, for convenience
24
  OUTPUT_VOCAB_SIZE = 16384 + 1 # encoded image token space + 1 for bos
25
- OUTPUT_LENGTH = 512 + 1 # number of encoded tokens + 1 for bos
26
  BOS_TOKEN_ID = 16384
27
  BASE_MODEL = 'facebook/bart-large-cnn' # we currently have issues with bart-large
28
 
 
22
 
23
  # Model hyperparameters, for convenience
24
  OUTPUT_VOCAB_SIZE = 16384 + 1 # encoded image token space + 1 for bos
25
+ OUTPUT_LENGTH = 256 + 1 # number of encoded tokens + 1 for bos
26
  BOS_TOKEN_ID = 16384
27
  BASE_MODEL = 'facebook/bart-large-cnn' # we currently have issues with bart-large
28