run-converted-model.sh 712 B

1234567891011121314151617181920212223242526
  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. MODEL_TESTING_PROMPT="${2:-"$MODEL_TESTING_PROMPT"}"
  6. if [ -z "$MODEL_TESTING_PROMPT"]; then
  7. MODEL_TESTING_PROMPT="Hello, my name is"
  8. fi
  9. # Final check if we have a model path
  10. if [ -z "$CONVERTED_MODEL" ]; then
  11. echo "Error: Model path must be provided either as:" >&2
  12. echo " 1. Command line argument" >&2
  13. echo " 2. CONVERTED_MODEL environment variable" >&2
  14. exit 1
  15. fi
  16. echo $CONVERTED_MODEL
  17. echo $MODEL_TESTING_PROMPT
  18. cmake --build ../../build --target llama-logits -j8
  19. ../../build/bin/llama-logits -m "$CONVERTED_MODEL" "$MODEL_TESTING_PROMPT"