|
@@ -1,5 +1,21 @@
|
|
|
#!/bin/bash
|
|
#!/bin/bash
|
|
|
|
|
|
|
|
|
|
+set -e
|
|
|
|
|
+
|
|
|
|
|
+# Parse command line arguments
|
|
|
|
|
+MMPROJ=""
|
|
|
|
|
+while [[ $# -gt 0 ]]; do
|
|
|
|
|
+ case $1 in
|
|
|
|
|
+ --mmproj)
|
|
|
|
|
+ MMPROJ="--mmproj"
|
|
|
|
|
+ shift
|
|
|
|
|
+ ;;
|
|
|
|
|
+ *)
|
|
|
|
|
+ shift
|
|
|
|
|
+ ;;
|
|
|
|
|
+ esac
|
|
|
|
|
+done
|
|
|
|
|
+
|
|
|
MODEL_NAME="${MODEL_NAME:-$(basename "$MODEL_PATH")}"
|
|
MODEL_NAME="${MODEL_NAME:-$(basename "$MODEL_PATH")}"
|
|
|
OUTPUT_DIR="${OUTPUT_DIR:-../../models}"
|
|
OUTPUT_DIR="${OUTPUT_DIR:-../../models}"
|
|
|
TYPE="${OUTTYPE:-f16}"
|
|
TYPE="${OUTTYPE:-f16}"
|
|
@@ -11,12 +27,20 @@ echo "Model name: ${MODEL_NAME}"
|
|
|
echo "Data type: ${TYPE}"
|
|
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}"
|
|
|
-python ../../convert_hf_to_gguf.py --verbose \
|
|
|
|
|
- ${MODEL_PATH} \
|
|
|
|
|
- --outfile ${CONVERTED_MODEL} \
|
|
|
|
|
- --outtype ${TYPE} \
|
|
|
|
|
- --metadata "${METADATA_OVERRIDE}"
|
|
|
|
|
|
|
+
|
|
|
|
|
+CMD_ARGS=("python" "../../convert_hf_to_gguf.py" "--verbose")
|
|
|
|
|
+CMD_ARGS+=("${MODEL_PATH}")
|
|
|
|
|
+CMD_ARGS+=("--outfile" "${CONVERTED_MODEL}")
|
|
|
|
|
+CMD_ARGS+=("--outtype" "${TYPE}")
|
|
|
|
|
+[[ -n "$METADATA_OVERRIDE" ]] && CMD_ARGS+=("--metadata" "${METADATA_OVERRIDE}")
|
|
|
|
|
+[[ -n "$MMPROJ" ]] && CMD_ARGS+=("${MMPROJ}")
|
|
|
|
|
+
|
|
|
|
|
+"${CMD_ARGS[@]}"
|
|
|
|
|
|
|
|
echo ""
|
|
echo ""
|
|
|
echo "The environment variable CONVERTED_MODEL can be set to this path using:"
|
|
echo "The environment variable CONVERTED_MODEL can be set to this path using:"
|
|
|
echo "export CONVERTED_MODEL=$(realpath ${CONVERTED_MODEL})"
|
|
echo "export CONVERTED_MODEL=$(realpath ${CONVERTED_MODEL})"
|
|
|
|
|
+if [[ -n "$MMPROJ" ]]; then
|
|
|
|
|
+ mmproj_file="${OUTPUT_DIR}/mmproj-$(basename "${CONVERTED_MODEL}")"
|
|
|
|
|
+ echo "The mmproj model was created in $(realpath "$mmproj_file")"
|
|
|
|
|
+fi
|