fix: CI

umbertogriffo · Dec 27, 2023 · 76465bf · 76465bf
1 parent c057eee
commit 76465bf
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
@@ -21,8 +21,8 @@ jobs:
         with:
           version: 1.7.0
           virtualenvs-in-project: true
-      # Installing sentence-transformers with pip to avoid poetry's issues in installing torch... (it doesn't install CUDA dependencies).
-      # Installing ctransformers and llama-cpp-python without NVIDIA CUDA acceleration.
+      # Installing torch (CPU) and sentence-transformers with pip to avoid poetry's issues in installing torch.
+      # Installing llama-cpp-python without NVIDIA CUDA acceleration.
       - name: Setup environment
         run: |
           poetry lock --check

diff --git a/Makefile b/Makefile
@@ -9,9 +9,8 @@ install:
 	mkdir -p .venv
 	poetry config virtualenvs.in-project true
 	poetry install
-	echo "Installing torch..."
+	echo "Installing torch and sentence-transformers with pip to avoid poetry's issues in installing torch... (it doesn't install CUDA dependencies)"
 	. .venv/bin/activate && pip3 install torch~=2.1.2 torchvision torchaudio
-	echo "Installing sentence-transformers with pip to avoid poetry's issues in installing torch... (it doesn't install CUDA dependencies)"
 	. .venv/bin/activate && pip3 install sentence-transformers~=2.2.2
 	echo "Installing llama-cpp-python with pip to get NVIDIA CUDA acceleration"
 	. .venv/bin/activate && CMAKE_ARGS="-DLLAMA_CUBLAS=on" pip3 install llama-cpp-python~=0.2.23