marstin's picture
[martin-dev] add demo v1 test
d425e71
architecture: blip2
model_path: Salesforce/blip2-opt-2.7b
model:
- torch_dtype: auto
output_db: output/blip2.db
input_dir: ./data/test-images/
prompt: "Describe the color in this image in one word."
modules:
- language_model.lm_head
- vision_model.post_layernorm