chris/MiniGPT-4
1
0
Fork 0
mirror of https://github.com/Vision-CAIR/MiniGPT-4.git synced 2025-04-05 18:40:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

update evaluation files

Browse Source
This commit is contained in:
junchen14 2023-11-01 12:18:06 +03:00
parent bf605d5d74
commit 748d18b5df
4 changed files with 4 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
eval_configs/minigptv2_benchmark_evaluation.yaml
View File

@ -66,7 +66,7 @@ evaluation_datasets:
eval_file_path: /path/to/eval/annotation/path eval_file_path: /path/to/eval/annotation/path
img_path: /path/to/eval/image/path img_path: /path/to/eval/image/path
max_new_tokens: 20 max_new_tokens: 20
batch_size: 10 batch_size: 100
run: run:
task: image_text_pretrain task: image_text_pretrain

2
eval_configs/minigptv2_eval.yaml
View File

@ -5,7 +5,7 @@ model:
end_sym: "</s>" end_sym: "</s>"
low_resource: True low_resource: True
prompt_template: '[INST] {} [/INST]' prompt_template: '[INST] {} [/INST]'
llama_model: "/ibex/project/c2133/llama_v2/llama-2-7b-chat-pytorch_update" llama_model: ""
ckpt: "" ckpt: ""
lora_r: 64 lora_r: 64
lora_alpha: 16 lora_alpha: 16

2
minigpt4/configs/models/minigpt_v2.yaml
View File

@ -11,7 +11,7 @@ model:
# generation configs # generation configs
prompt: "" prompt: ""
llama_model: "/ibex/project/c2133/llama_v2/llama-2-7b-chat-pytorch" llama_model: ""
lora_r: 64 lora_r: 64
lora_alpha: 16 lora_alpha: 16

5
minigpt4/models/minigpt_base.py
View File

@ -10,11 +10,6 @@ from minigpt4.models.base_model import BaseModel
from transformers import StoppingCriteria, StoppingCriteriaList from transformers import StoppingCriteria, StoppingCriteriaList
from minigpt4.conversation.conversation import StoppingCriteriaSub from minigpt4.conversation.conversation import StoppingCriteriaSub
<<<<<<< HEAD
=======
>>>>>>> upstream/main
class MiniGPTBase(BaseModel): class MiniGPTBase(BaseModel):
""" """