noamrot
/

FuseCap_Image_Captioning

image-captioning

Model card Files Files and versions

noamrot commited on Jun 1, 2023

Commit

b74ad6f

·

1 Parent(s): 966304f

fixed beam number

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -53,7 +53,7 @@
     "no_repeat_ngram_size": 0,
     "num_attention_heads": 12,
     "num_beam_groups": 1,
-    "num_beams": 1,
     "num_hidden_layers": 12,
     "num_return_sequences": 1,
     "output_attentions": false,

     "no_repeat_ngram_size": 0,
     "num_attention_heads": 12,
     "num_beam_groups": 1,
+    "num_beams": 3,
     "num_hidden_layers": 12,
     "num_return_sequences": 1,
     "output_attentions": false,