tarekziade
/

vit-base-patch16-224-in21k-distilgpt2

vision-encoder-decoder

image-text-to-text

image-captioning

Inference Endpoints

Model card Files Files and versions Community

tarekziade commited on Jun 4

Commit

aaa7cc5

•

1 Parent(s): a12acc3

Update generation_config.json

Files changed (1) hide show

generation_config.json +7 -3

generation_config.json CHANGED Viewed

@@ -1,5 +1,9 @@
 {
   "bos_token_id": 50256,
-  "eos_token_id": 50256,
-  "transformers_version": "4.33.2"
-}

 {
   "bos_token_id": 50256,
+  "max_length": 50,
+  "num_beams": 2,
+  "early_stopping": true,
+  "no_repeat_ngram_size": 2,
+  "pad_token_id": 50256,
+  "eos_token_id": 50256
+}