libs/model_list.py

model_list = {'blip2' : ["Salesforce/blip2-opt-2.7b", "Salesforce/blip2-flan-t5-xxl"],
              'blip' : ["Salesforce/blip-vqa-base"],
              'vit_gpt2':["nlpconnect/vit-gpt2-image-captioning"], 
              'InstructBlip': ["Salesforce/instructblip-vicuna-7b"]
              }