From f50cc767318eb3dde362b5788e040804228011f3 Mon Sep 17 00:00:00 2001 From: Alexander Belikov Date: Mon, 2 Feb 2026 22:36:50 +0100 Subject: [PATCH] fixin bugs --- install-cuda121.sh | 9 +++++++++ run/loop.embed.kb.corpus.sh | 2 +- 2 files changed, 10 insertions(+), 1 deletion(-) create mode 100755 install-cuda121.sh 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 ;;