mirror of
https://github.com/Vision-CAIR/MiniGPT-4.git
synced 2025-04-05 02:20:47 +00:00
update default ckpt
This commit is contained in:
parent
07c5d9f73a
commit
a6e9a4c2bc
2
.gitignore
vendored
2
.gitignore
vendored
@ -183,3 +183,5 @@ dataset/Evaluation.md
|
||||
jupyter_notebook.slurm
|
||||
MVTEC/
|
||||
MVTEC_det/
|
||||
*.pth
|
||||
log.*
|
25
eval_configs/minigptv2_eval_mvtec.yaml
Normal file
25
eval_configs/minigptv2_eval_mvtec.yaml
Normal file
@ -0,0 +1,25 @@
|
||||
model:
|
||||
arch: minigpt_v2
|
||||
model_type: pretrain
|
||||
max_txt_len: 500
|
||||
end_sym: "</s>"
|
||||
low_resource: True
|
||||
prompt_template: '[INST] {} [/INST]'
|
||||
# ckpt: "./minigpt4/mvtec_outputs/20250112105/checkpoint_9.pth"
|
||||
ckpt: "./ckpt/checkpoint_stage3.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
|
@ -11,7 +11,7 @@ model:
|
||||
# generation configs
|
||||
prompt: ""
|
||||
|
||||
llama_model: "please set this value to the path of llama2-chat-7b"
|
||||
llama_model: "meta-llama/Llama-2-7b-chat-hf"
|
||||
lora_r: 64
|
||||
lora_alpha: 16
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user