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

30 lines
539 B
YAML
Raw Normal View History

2023-04-16 22:04:16 +00:00
model:
arch: mini_gpt4
# vit encoder
image_size: 512
2023-04-16 22:04:16 +00:00
drop_path_rate: 0
use_grad_checkpoint: False
vit_precision: "fp32"
2023-04-16 22:04:16 +00:00
freeze_vit: True
2023-08-28 18:26:00 +00:00
has_qformer: False
2023-04-16 22:04:16 +00:00
# generation configs
prompt: ""
llama_model: "Llama-2-7b-chat"
2023-08-28 18:26:00 +00:00
2023-04-16 22:04:16 +00:00
preprocess:
vis_processor:
train:
name: "blip2_image_train"
image_size: 512
2023-04-16 22:04:16 +00:00
eval:
name: "blip2_image_eval"
image_size: 512
2023-04-16 22:04:16 +00:00
text_processor:
train:
name: "blip_caption"
eval:
name: "blip_caption"