mirror of
https://github.com/Vision-CAIR/MiniGPT-4.git
synced 2025-04-05 18:40:46 +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
|
jupyter_notebook.slurm
|
||||||
MVTEC/
|
MVTEC/
|
||||||
MVTEC_det/
|
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
|
# generation configs
|
||||||
prompt: ""
|
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_r: 64
|
||||||
lora_alpha: 16
|
lora_alpha: 16
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user