MiniGPT-4/eval_configs/minigptv2_eval.yaml

25 lines
458 B
YAML
Raw Normal View History

2023-10-12 21:14:34 +00:00
model:
arch: minigpt_v2
model_type: pretrain
max_txt_len: 160
end_sym: "</s>"
low_resource: True
prompt_template: '[INST] {} [/INST]'
2023-10-13 14:11:11 +00:00
ckpt: 'please set this value to the path of pretrained checkpoint'
2023-10-12 21:14:34 +00:00
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