Skip to content

Commit 3bd2c7c

Browse files
authored
docker : add finetune option (ggml-org#4211)
1 parent bde629b commit 3bd2c7c

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

.devops/tools.sh

+4
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,8 @@ elif [[ "$arg1" == '--quantize' || "$arg1" == '-q' ]]; then
1313
./quantize "$@"
1414
elif [[ "$arg1" == '--run' || "$arg1" == '-r' ]]; then
1515
./main "$@"
16+
elif [[ "$arg1" == '--finetune' || "$arg1" == '-f' ]]; then
17+
./finetune "$@"
1618
elif [[ "$arg1" == '--all-in-one' || "$arg1" == '-a' ]]; then
1719
echo "Converting PTH to GGML..."
1820
for i in `ls $1/$2/ggml-model-f16.bin*`; do
@@ -34,6 +36,8 @@ else
3436
echo " ex: --outtype f16 \"/models/7B/\" "
3537
echo " --quantize (-q): Optimize with quantization process ggml"
3638
echo " ex: \"/models/7B/ggml-model-f16.bin\" \"/models/7B/ggml-model-q4_0.bin\" 2"
39+
echo " --finetune (-f): Run finetune command to create a lora finetune of the model"
40+
echo " See documentation for finetune for command-line parameters"
3741
echo " --all-in-one (-a): Execute --convert & --quantize"
3842
echo " ex: \"/models/\" 7B"
3943
echo " --server (-s): Run a model on the server"

0 commit comments

Comments
 (0)