run-converted-model-embeddings-logits.sh 636 B

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