mirror of
https://github.com/ggml-org/llama.cpp.git
synced 2026-08-06 18:10:51 -05:00
76c956c137
* Add delete-files option to split parameters Added a new option to delete split files during execution to free up disk space. * Add test for delete files on merge option * Fix tests * Update tools/gguf-split/gguf-split.cpp Co-authored-by: Sigbjørn Skjæret <sigbjorn.skjaeret@huggingface.co> * Update tools/gguf-split/gguf-split.cpp Co-authored-by: Sigbjørn Skjæret <sigbjorn.skjaeret@huggingface.co> * Uncomment tests * Improvements to address PR comments * Fix formatting * Fix formatting * Rename --delete-files to --delete-splits * Comment tests * Move delete inside loop * style cleanup --------- Co-authored-by: Sigbjørn Skjæret <sigbjorn.skjaeret@huggingface.co>
110 lines
3.0 KiB
Bash
Executable File
110 lines
3.0 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
|
|
set -eu
|
|
|
|
if [ $# -lt 1 ]
|
|
then
|
|
echo "usage: $0 path_to_build_binary [path_to_temp_folder]"
|
|
echo "example: $0 ../../build/bin ../../tmp"
|
|
exit 1
|
|
fi
|
|
|
|
if [ $# -gt 1 ]
|
|
then
|
|
TMP_DIR=$2
|
|
else
|
|
TMP_DIR=/tmp
|
|
fi
|
|
|
|
set -x
|
|
|
|
SPLIT=$1/llama-gguf-split
|
|
MAIN=$1/llama-completion
|
|
WORK_PATH=$TMP_DIR/gguf-split
|
|
ROOT_DIR=$(realpath $(dirname $0)/../../)
|
|
|
|
mkdir -p "$WORK_PATH"
|
|
|
|
# Clean up in case of previously failed test
|
|
rm -f $WORK_PATH/ggml-model-split*.gguf $WORK_PATH/ggml-model-merge*.gguf
|
|
|
|
# 1. Get a model
|
|
(
|
|
cd $WORK_PATH
|
|
"$ROOT_DIR"/scripts/hf.sh --repo ggml-org/Qwen3-0.6B-GGUF --file Qwen3-0.6B-Q8_0.gguf
|
|
)
|
|
echo PASS
|
|
|
|
# 2. Split with max tensors strategy
|
|
$SPLIT --split-max-tensors 28 $WORK_PATH/Qwen3-0.6B-Q8_0.gguf $WORK_PATH/ggml-model-split
|
|
echo PASS
|
|
echo
|
|
|
|
# 2b. Test the sharded model is loading properly
|
|
$MAIN -no-cnv --model $WORK_PATH/ggml-model-split-00001-of-00012.gguf -p "I believe the meaning of life is" --n-predict 32
|
|
echo PASS
|
|
echo
|
|
|
|
# 3. Merge
|
|
$SPLIT --merge $WORK_PATH/ggml-model-split-00001-of-00012.gguf $WORK_PATH/ggml-model-merge.gguf
|
|
echo PASS
|
|
echo
|
|
|
|
# 3b. Test the merged model is loading properly
|
|
$MAIN -no-cnv --model $WORK_PATH/ggml-model-merge.gguf -p "I believe the meaning of life is" --n-predict 32
|
|
echo PASS
|
|
echo
|
|
|
|
# 4. Split with no tensors in the first split
|
|
$SPLIT --split-max-tensors 32 --no-tensor-first-split $WORK_PATH/ggml-model-merge.gguf $WORK_PATH/ggml-model-split-32-tensors
|
|
echo PASS
|
|
echo
|
|
|
|
# 4b. Test the sharded model is loading properly
|
|
$MAIN -no-cnv --model $WORK_PATH/ggml-model-split-32-tensors-00001-of-00011.gguf -p "I believe the meaning of life is" --n-predict 32
|
|
echo PASS
|
|
echo
|
|
|
|
# 5. Merge
|
|
#$SPLIT --merge $WORK_PATH/ggml-model-split-32-tensors-00001-of-00011.gguf $WORK_PATH/ggml-model-merge-2.gguf
|
|
#echo PASS
|
|
#echo
|
|
|
|
# 5b. Test the merged model is loading properly
|
|
#$MAIN -no-cnv --model $WORK_PATH/ggml-model-merge-2.gguf -p "I believe the meaning of life is" --n-predict 32
|
|
#echo PASS
|
|
#echo
|
|
|
|
# 6. Split with size strategy
|
|
$SPLIT --split-max-size 500M $WORK_PATH/ggml-model-merge.gguf $WORK_PATH/ggml-model-split-500M
|
|
echo PASS
|
|
echo
|
|
|
|
# 6b. Test the sharded model is loading properly
|
|
$MAIN -no-cnv --model $WORK_PATH/ggml-model-split-500M-00001-of-00002.gguf -p "I believe the meaning of life is" --n-predict 32
|
|
echo PASS
|
|
echo
|
|
|
|
# 7. Merge with delete splits
|
|
#for i in $(seq -w 1 11); do
|
|
# cp "$WORK_PATH/ggml-model-split-32-tensors-000${i}-of-00011.gguf" "$WORK_PATH/ggml-model-split-32-tensors-copy-000${i}-of-00011.gguf"
|
|
#done
|
|
#$SPLIT --merge --delete-splits $WORK_PATH/ggml-model-split-32-tensors-copy-00001-of-00011.gguf $WORK_PATH/ggml-model-merge-3.gguf
|
|
#echo PASS
|
|
#echo
|
|
|
|
# 7b. Test the merged model is loading properly
|
|
#$MAIN -no-cnv --model $WORK_PATH/ggml-model-merge-3.gguf -p "I believe the meaning of life is" --n-predict 32
|
|
#echo PASS
|
|
#echo
|
|
|
|
# 7c. Test the files were deleted
|
|
#for i in $(seq -w 1 11); do
|
|
# test ! -f "$WORK_PATH/ggml-model-split-32-tensors-copy-000${i}-of-00011.gguf"
|
|
#done
|
|
#echo PASS
|
|
#echo
|
|
|
|
# Clean up
|
|
rm -f $WORK_PATH/ggml-model-split*.gguf $WORK_PATH/ggml-model-merge*.gguf
|