feat: Support vicuna-v1.5 and WizardLM-v1.2

This commit is contained in:
FangYin Cheng
2023-08-03 14:13:50 +08:00
parent 1388f33ddc
commit a4574aa614
11 changed files with 140 additions and 49 deletions

View File

@@ -4,10 +4,15 @@ SCRIPT_LOCATION=$0
cd "$(dirname "$SCRIPT_LOCATION")"
WORK_DIR=$(pwd)
if [[ " $* " == *" --help "* ]] || [[ " $* " == *" -h "* ]]; then
bash $WORK_DIR/base/build_image.sh "$@"
exit 0
fi
bash $WORK_DIR/base/build_image.sh "$@"
if [ 0 -ne $? ]; then
ehco "Error: build base image failed"
echo "Error: build base image failed"
exit 1
fi