Skip to content

Commit b5c8546

Browse files
authored
Docker: change to calling convert.py (#1641)
Deprecation disclaimer was added to convert-pth-to-ggml.py
1 parent 136476e commit b5c8546

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

.devops/tools.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@ shift
1111
arg2="$@"
1212

1313
if [[ $arg1 == '--convert' || $arg1 == '-c' ]]; then
14-
python3 ./convert-pth-to-ggml.py $arg2
14+
python3 ./convert.py $arg2
1515
elif [[ $arg1 == '--quantize' || $arg1 == '-q' ]]; then
1616
./quantize $arg2
1717
elif [[ $arg1 == '--run' || $arg1 == '-r' ]]; then
@@ -32,7 +32,7 @@ else
3232
echo " --run (-r): Run a model previously converted into ggml"
3333
echo " ex: -m /models/7B/ggml-model-q4_0.bin -p \"Building a website can be done in 10 simple steps:\" -n 512"
3434
echo " --convert (-c): Convert a llama model into ggml"
35-
echo " ex: \"/models/7B/\" 1"
35+
echo " ex: --outtype f16 \"/models/7B/\" "
3636
echo " --quantize (-q): Optimize with quantization process ggml"
3737
echo " ex: \"/models/7B/ggml-model-f16.bin\" \"/models/7B/ggml-model-q4_0.bin\" 2"
3838
echo " --all-in-one (-a): Execute --convert & --quantize"

convert-pth-to-ggml.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,9 @@
44

55
import convert
66

7-
parser = argparse.ArgumentParser(description='Convert a LLaMA model checkpoint to a ggml compatible file')
7+
parser = argparse.ArgumentParser(
8+
description="""[DEPRECATED - use `convert.py` instead]
9+
Convert a LLaMA model checkpoint to a ggml compatible file""")
810
parser.add_argument('dir_model', help='directory containing the model checkpoint')
911
parser.add_argument('ftype', help='file type (0: float32, 1: float16)', type=int, choices=[0, 1], default=1)
1012
args = parser.parse_args()

0 commit comments

Comments
 (0)