Salesforce
/

blip-image-captioning-base

image-text-to-text

image-captioning

Inference Endpoints

Model card Files Files and versions Community

Update config.json

#13

by ybelkada - opened Apr 6, 2023

base: refs/heads/main

←

from: refs/pr/13

Discussion Files changed

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -51,7 +51,7 @@
     "min_length": 0,
     "model_type": "blip_text_model",
     "no_repeat_ngram_size": 0,
-    "num_attention_heads": 8,
     "num_beam_groups": 1,
     "num_beams": 1,
     "num_hidden_layers": 12,

     "min_length": 0,
     "model_type": "blip_text_model",
     "no_repeat_ngram_size": 0,
+    "num_attention_heads": 12,
     "num_beam_groups": 1,
     "num_beams": 1,
     "num_hidden_layers": 12,