model-conversion : add debug option to conversion script (#19265)
This commit adds a debug option to the model conversion script to enable using the Python debugger (pdb) during model conversion. The motivation for this is that I've found myself adding this a few times now and it would be quicker to have this flag as an option and a makefile target/recipe for it.
This commit is contained in:
parent
59377a6c87
commit
6156ae5111
|
|
@ -33,11 +33,14 @@ DEVICE ?= auto
|
||||||
causal-convert-model-bf16: OUTTYPE=bf16
|
causal-convert-model-bf16: OUTTYPE=bf16
|
||||||
causal-convert-model-bf16: causal-convert-model
|
causal-convert-model-bf16: causal-convert-model
|
||||||
|
|
||||||
|
causal-convert-model-debug: DEBUG=--debug
|
||||||
|
causal-convert-model-debug: causal-convert-model
|
||||||
|
|
||||||
causal-convert-model:
|
causal-convert-model:
|
||||||
$(call validate_model_path,causal-convert-model)
|
$(call validate_model_path,causal-convert-model)
|
||||||
@MODEL_NAME="$(MODEL_NAME)" OUTTYPE="$(OUTTYPE)" MODEL_PATH="$(MODEL_PATH)" \
|
@MODEL_NAME="$(MODEL_NAME)" OUTTYPE="$(OUTTYPE)" MODEL_PATH="$(MODEL_PATH)" \
|
||||||
METADATA_OVERRIDE="$(METADATA_OVERRIDE)" \
|
METADATA_OVERRIDE="$(METADATA_OVERRIDE)" \
|
||||||
./scripts/causal/convert-model.sh
|
./scripts/causal/convert-model.sh $(DEBUG)
|
||||||
|
|
||||||
causal-convert-mm-model-bf16: OUTTYPE=bf16
|
causal-convert-mm-model-bf16: OUTTYPE=bf16
|
||||||
causal-convert-mm-model-bf16: MM_OUTTYPE=f16
|
causal-convert-mm-model-bf16: MM_OUTTYPE=f16
|
||||||
|
|
|
||||||
|
|
@ -4,12 +4,17 @@ set -e
|
||||||
|
|
||||||
# Parse command line arguments
|
# Parse command line arguments
|
||||||
MMPROJ=""
|
MMPROJ=""
|
||||||
|
DEBUG=""
|
||||||
while [[ $# -gt 0 ]]; do
|
while [[ $# -gt 0 ]]; do
|
||||||
case $1 in
|
case $1 in
|
||||||
--mmproj)
|
--mmproj)
|
||||||
MMPROJ="--mmproj"
|
MMPROJ="--mmproj"
|
||||||
shift
|
shift
|
||||||
;;
|
;;
|
||||||
|
--debug)
|
||||||
|
DEBUG="1"
|
||||||
|
shift
|
||||||
|
;;
|
||||||
*)
|
*)
|
||||||
shift
|
shift
|
||||||
;;
|
;;
|
||||||
|
|
@ -28,7 +33,12 @@ echo "Data type: ${TYPE}"
|
||||||
echo "Converted model path:: ${CONVERTED_MODEL}"
|
echo "Converted model path:: ${CONVERTED_MODEL}"
|
||||||
echo "Metadata override: ${METADATA_OVERRIDE}"
|
echo "Metadata override: ${METADATA_OVERRIDE}"
|
||||||
|
|
||||||
CMD_ARGS=("python" "../../convert_hf_to_gguf.py" "--verbose")
|
if [[ -n "$DEBUG" ]]; then
|
||||||
|
CMD_ARGS=("python" "-m" "pdb")
|
||||||
|
else
|
||||||
|
CMD_ARGS=("python")
|
||||||
|
fi
|
||||||
|
CMD_ARGS+=("../../convert_hf_to_gguf.py" "--verbose")
|
||||||
CMD_ARGS+=("${MODEL_PATH}")
|
CMD_ARGS+=("${MODEL_PATH}")
|
||||||
CMD_ARGS+=("--outfile" "${CONVERTED_MODEL}")
|
CMD_ARGS+=("--outfile" "${CONVERTED_MODEL}")
|
||||||
CMD_ARGS+=("--outtype" "${TYPE}")
|
CMD_ARGS+=("--outtype" "${TYPE}")
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue