Start work on llm_engine
This commit is contained in:
25
test.sh
Normal file
25
test.sh
Normal file
@@ -0,0 +1,25 @@
|
||||
#!/bin/bash
|
||||
|
||||
# Continue on error
|
||||
set -e
|
||||
|
||||
# Build with progress output and capture the tag
|
||||
TAG=$( \
|
||||
docker build \
|
||||
--target test \
|
||||
. \
|
||||
2>&1 | tee /dev/tty | grep "writing image" | cut -d' ' -f4 \
|
||||
)
|
||||
|
||||
# Exit if tag is empty
|
||||
[ -z "$TAG" ] && exit 1
|
||||
|
||||
# Run tests
|
||||
docker run \
|
||||
--rm \
|
||||
--gpus=all \
|
||||
-v /$(pwd)/model/:/root/model/ \
|
||||
$TAG
|
||||
|
||||
# Clean up image
|
||||
[ ! -z "$TAG" ] && docker rmi $TAG
|
||||
Reference in New Issue
Block a user