Merge pull request #459 from julien-blanchon/patch-3

Add BLIP ImageCaption model
2024-03-06 22:35:41 +09:00
parent e47b098e3f fafdc3d4b7
commit 7e4bdb6e55
1 changed files with 9 additions and 0 deletions
--- a/model-list.json
+++ b/model-list.json
@@ -2171,6 +2171,15 @@
      "url": "https://huggingface.co/public-data/insightface/resolve/main/models/buffalo_l/w600k_r50.onnx"
    },
    {
+        "name": "BLIP ImageCaption (COCO) w/ ViT-B and CapFilt-L",
+        "type": "BLIP_MODEL",
+        "base": "blip_model",
+        "save_path": "blip",
+        "description": "BLIP ImageCaption (COCO) w/ ViT-B and CapFilt-L",
+        "reference": "https://github.com/salesforce/BLIP",
+        "filename": "model_base_capfilt_large.pth",
+        "url": "https://storage.googleapis.com/sfr-vision-language-research/BLIP/models/model_base_capfilt_large.pth"
+    },
        "name": "GroundingDINO SwinT OGC - Model",
        "type": "GroundingDINO",
        "base": "DINO",