MiniGPT-4/minigpt4/configs/models/bindgpt4.yaml

30 lines
530 B
YAML
Raw Normal View History

2023-05-18 05:52:05 +00:00
model:
arch: bind_gpt4
# Imagebind
freeze_imagebind: True
# Q-Former
freeze_qformer: True
num_query_token: 32
# Vicuna
llama_model: "/path/to/vicuna/weights/"
# generation configs
prompt: ""
preprocess:
vis_processor:
train:
name: "imagebind_vision_train"
image_size: 224
eval:
name: "imagebind_vision_eval"
image_size: 224
text_processor:
train:
name: "imagebind_caption"
eval:
name: "imagebind_caption"