Skip to content

Commit

Permalink
[Docker] fix tools.sh argument passing. (#2884)
Browse files Browse the repository at this point in the history
* [Docker] fix tools.sh argument passing.

This should allow passing multiple arguments to containers with
the full image that are using the tools.sh frontend.

Fix from #2535 (comment)
  • Loading branch information
SlyEcho authored Aug 30, 2023
1 parent b532a69 commit 71d6975
Showing 1 changed file with 4 additions and 7 deletions.
11 changes: 4 additions & 7 deletions .devops/tools.sh
Original file line number Diff line number Diff line change
Expand Up @@ -7,15 +7,12 @@ arg1="$1"
# Shift the arguments to remove the first one
shift

# Join the remaining arguments into a single string
arg2="$@"

if [[ "$arg1" == '--convert' || "$arg1" == '-c' ]]; then
python3 ./convert.py "$arg2"
python3 ./convert.py "$@"
elif [[ "$arg1" == '--quantize' || "$arg1" == '-q' ]]; then
./quantize "$arg2"
./quantize "$@"
elif [[ "$arg1" == '--run' || "$arg1" == '-r' ]]; then
./main "$arg2"
./main "$@"
elif [[ "$arg1" == '--all-in-one' || "$arg1" == '-a' ]]; then
echo "Converting PTH to GGML..."
for i in `ls $1/$2/ggml-model-f16.bin*`; do
Expand All @@ -27,7 +24,7 @@ elif [[ "$arg1" == '--all-in-one' || "$arg1" == '-a' ]]; then
fi
done
elif [[ "$arg1" == '--server' || "$arg1" == '-s' ]]; then
./server "$arg2"
./server "$@"
else
echo "Unknown command: $arg1"
echo "Available commands: "
Expand Down

0 comments on commit 71d6975

Please sign in to comment.