support caption model

Author: Enderfga

llama_adapter_v2_multimodal/demo.py

@@ -7,7 +7,7 @@
 
 llama_dir = "/path/to/LLaMA/"
 
-# choose from BIAS-7B, LORA-BIAS-7B
+# choose from BIAS-7B, LORA-BIAS-7B, CAPTION-7B.pth
 model, preprocess = llama.load("BIAS-7B", llama_dir, device)
 model.eval()
 

llama_adapter_v2_multimodal/llama/llama_adapter.py

@@ -278,6 +278,7 @@ def generate(
 _MODELS = {
     "BIAS-7B": "https://github.com/OpenGVLab/LLaMA-Adapter/releases/download/v.2.0.0/7fa55208379faf2dd862565284101b0e4a2a72114d6490a95e432cf9d9b6c813_BIAS-7B.pth",
     "LORA-BIAS-7B": "https://github.com/OpenGVLab/LLaMA-Adapter/releases/download/v.2.0.0/1bcbffc43484332672092e0024a8699a6eb5f558161aebf98a7c6b1db67224d1_LORA-BIAS-7B.pth",
+    "CAPTION-7B": "https://github.com/OpenGVLab/LLaMA-Adapter/releases/download/v.2.0.0/5088aeb63a89746b90bcfd5cb819e1c7411b2771b267c6d131ce73e250a8abf0_CAPTION-7B.pth",
     # "LORA16-7B": "",
     # "PARTIAL-7B": ""
 }