diff --git a/install-cuda121.sh b/install-cuda121.sh new file mode 100755 index 0000000..4e4d7e2 --- /dev/null +++ b/install-cuda121.sh @@ -0,0 +1,9 @@ +#!/usr/bin/env bash +set -e + +pip install torch torchvision torchaudio \ + --index-url https://download.pytorch.org/whl/cu121 + +pip install cupy-cuda12x + +pip install -e . diff --git a/run/loop.embed.kb.corpus.sh b/run/loop.embed.kb.corpus.sh index 7b34a67..19b2afe 100644 --- a/run/loop.embed.kb.corpus.sh +++ b/run/loop.embed.kb.corpus.sh @@ -19,7 +19,7 @@ while [[ $# -gt 0 ]]; do OUTPUT_PARQUET_PREFIX="$2" shift 2 ;; - --properties-txt-path) + --kb-csv-path) KB_CSV_PATH="$2" shift 2 ;;