MiniGPT-v2 / eval_configs /minigptv2_eval.yaml
Vision-CAIR's picture
Update eval_configs/minigptv2_eval.yaml
15309b3
model:
arch: minigpt_v2
model_type: pretrain
max_txt_len: 160
end_sym: "</s>"
low_resource: True
prompt_template: '[INST] {} [/INST]'
ckpt: 'minigptv2_checkpoint.pth'
lora_r: 64
lora_alpha: 16
datasets:
cc_sbu_align:
vis_processor:
train:
name: "blip2_image_eval"
image_size: 448
text_processor:
train:
name: "blip_caption"
run:
task: image_text_pretrain