chris/MiniGPT-4
1
0
Fork 0
mirror of https://github.com/Vision-CAIR/MiniGPT-4.git synced 2025-04-06 19:10:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

add demo.sh

Browse Source
This commit is contained in:
wacdev 2023-04-30 13:54:09 +08:00
parent 407cb8420a
commit e2c0d3f82a
No known key found for this signature in database
GPG Key ID: 5473A7C3DEAFC58D
3 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

0
demo.py Normal file → Executable file
View File

7
demo.sh Executable file
View File

@ -0,0 +1,7 @@
#!/usr/bin/env bash
DIR=$(realpath $0) && DIR=${DIR%/*}
cd $DIR
set -ex
python demo.py --cfg-path $DIR/eval_configs/minigpt4_eval.yaml

2
eval_configs/minigpt4_eval.yaml
View File

@ -8,7 +8,7 @@ model:
low_resource: True low_resource: True
prompt_path: "prompts/alignment.txt" prompt_path: "prompts/alignment.txt"
prompt_template: '###Human: {} ###Assistant: ' prompt_template: '###Human: {} ###Assistant: '
ckpt: '/path/to/pretrained/ckpt/' ckpt: ./model/pretrained_minigpt4.pth # /path/to/pretrained/ckpt/
datasets: datasets: