run-embedding-server.sh 646 B

123456789101112131415161718192021222324252627
  1. #!/usr/bin/env bash
  2. set -e
  3. #
  4. # First try command line argument, then environment variable, then file
  5. CONVERTED_MODEL="${1:-"$CONVERTED_MODEL"}"
  6. BUILD_DIR="${2:-"$BUILD_DIR"}"
  7. # Final check if we have a model path
  8. if [ -z "$CONVERTED_MODEL" ]; then
  9. echo "Error: Model path must be provided either as:" >&2
  10. echo " 1. Command line argument" >&2
  11. echo " 2. CONVERTED_MODEL environment variable" >&2
  12. exit 1
  13. fi
  14. if [ -z "$BUILD_DIR" ]; then
  15. BUILD_DIR="../../build"
  16. fi
  17. echo $CONVERTED_MODEL
  18. cmake --build $BUILD_DIR --target llama-server
  19. ${BUILD_DIR}/bin/llama-server -m $CONVERTED_MODEL \
  20. --embedding \
  21. --pooling none