From 3585f804144ca553171150518435af9be4661b7d Mon Sep 17 00:00:00 2001 From: Adam Wilson Date: Tue, 19 Aug 2025 19:51:12 -0600 Subject: [PATCH] fix dependencies --- requirements.txt | 4 +++- run.sh | 6 +++--- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/requirements.txt b/requirements.txt index 29f9c568b..b620a6755 100644 --- a/requirements.txt +++ b/requirements.txt @@ -201,7 +201,9 @@ threadpoolctl==3.6.0 tiktoken==0.9.0 tokenizers==0.21.1 tomli==2.2.1 -torch==2.7.0 +torch==2.8.0+cpu +torchaudio==2.8.0+cpu +torchvision==0.23.0+cpu tqdm==4.67.1 transformers==4.51.3 triton==3.3.0 diff --git a/run.sh b/run.sh index 05214e39b..0186239bb 100755 --- a/run.sh +++ b/run.sh @@ -24,14 +24,14 @@ if [ "$LOCAL" = true ]; then fi # the ONNX model/data require git Large File System support -# git lfs install +git lfs install # install Python dependencies -# pip install -r ./requirements.txt +pip install -r ./requirements.txt # environment variables -export GARAK_PROBE='latentinjection.LatentInjectionResume' +# export GARAK_PROBE='latentinjection.LatentInjectionResume' # export GARAK_PROBE='latentinjection.LatentInjectionTranslationEnFr' # export GARAK_PROBE='latentinjection.LatentInjectionTranslationEnZh' # export GARAK_PROBE='latentinjection.LatentWhoisSnippet'