From e2c0d3f82adcb561efbb1c55a20c420eaa718429 Mon Sep 17 00:00:00 2001 From: wacdev Date: Sun, 30 Apr 2023 13:54:09 +0800 Subject: [PATCH] add demo.sh --- demo.py | 0 demo.sh | 7 +++++++ eval_configs/minigpt4_eval.yaml | 2 +- 3 files changed, 8 insertions(+), 1 deletion(-) mode change 100644 => 100755 demo.py create mode 100755 demo.sh diff --git a/demo.py b/demo.py old mode 100644 new mode 100755 diff --git a/demo.sh b/demo.sh new file mode 100755 index 0000000..78c95b1 --- /dev/null +++ b/demo.sh @@ -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 diff --git a/eval_configs/minigpt4_eval.yaml b/eval_configs/minigpt4_eval.yaml index f9e55a3..3c7df35 100644 --- a/eval_configs/minigpt4_eval.yaml +++ b/eval_configs/minigpt4_eval.yaml @@ -8,7 +8,7 @@ model: low_resource: True prompt_path: "prompts/alignment.txt" prompt_template: '###Human: {} ###Assistant: ' - ckpt: '/path/to/pretrained/ckpt/' + ckpt: ./model/pretrained_minigpt4.pth # /path/to/pretrained/ckpt/ datasets: