convert-model.sh 699 B

12345678910111213141516171819202122
  1. #!/bin/bash
  2. set -e
  3. MODEL_NAME="${MODEL_NAME:-$(basename "$EMBEDDING_MODEL_PATH")}"
  4. OUTPUT_DIR="${OUTPUT_DIR:-../../models}"
  5. TYPE="${OUTTYPE:-f16}"
  6. METADATA_OVERRIDE="${METADATA_OVERRIDE:-}"
  7. CONVERTED_MODEL="${OUTPUT_DIR}/${MODEL_NAME}.gguf"
  8. echo "Model path: ${EMBEDDING_MODEL_PATH}"
  9. echo "Model name: ${MODEL_NAME}"
  10. echo "Data type: ${TYPE}"
  11. echo "Converted model path:: ${CONVERTED_MODEL}"
  12. python ../../convert_hf_to_gguf.py --verbose \
  13. ${EMBEDDING_MODEL_PATH} \
  14. --outfile ${CONVERTED_MODEL} \
  15. --outtype ${TYPE}
  16. echo ""
  17. echo "The environment variable CONVERTED_EMBEDDING MODEL can be set to this path using:"
  18. echo "export CONVERTED_EMBEDDING_MODEL=$(realpath ${CONVERTED_MODEL})"