From a6e9a4c2bc9315591b72f61d8b68e970bed87f3a Mon Sep 17 00:00:00 2001
From: ThuanNaN <duongthuan1445@gmail.com>
Date: Sun, 12 Jan 2025 18:46:18 +0700
Subject: [PATCH] update default ckpt

---
 .gitignore                              |  2 ++
 eval_configs/minigptv2_eval_mvtec.yaml  | 25 +++++++++++++++++++++++++
 minigpt4/configs/models/minigpt_v2.yaml |  2 +-
 3 files changed, 28 insertions(+), 1 deletion(-)
 create mode 100644 eval_configs/minigptv2_eval_mvtec.yaml

diff --git a/.gitignore b/.gitignore
index 4b91a22..0815ff0 100755
--- a/.gitignore
+++ b/.gitignore
@@ -183,3 +183,5 @@ dataset/Evaluation.md
 jupyter_notebook.slurm
 MVTEC/
 MVTEC_det/
+*.pth
+log.*
\ No newline at end of file
diff --git a/eval_configs/minigptv2_eval_mvtec.yaml b/eval_configs/minigptv2_eval_mvtec.yaml
new file mode 100644
index 0000000..7539714
--- /dev/null
+++ b/eval_configs/minigptv2_eval_mvtec.yaml
@@ -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
diff --git a/minigpt4/configs/models/minigpt_v2.yaml b/minigpt4/configs/models/minigpt_v2.yaml
index 1d85d20..c649eb6 100755
--- a/minigpt4/configs/models/minigpt_v2.yaml
+++ b/minigpt4/configs/models/minigpt_v2.yaml
@@ -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