fix: import sentence_transfomers on pure cpu was failing (#429)
Co-authored-by: Rafael Uzarowski <uzarowski.rafael@proton.me>
ehl0wr0ld committed
Jun 3, 2025 at 16:28 UTC
f37974c3b14aa293802889b7fe2469fd724c2357
2 files changed
+6
-3
docker/base/fs/ins/install_python.sh
+1
-1
@@ -46,7 +46,7 @@ source /opt/venv/bin/activate
46
# upgrade pip and install static packages
47
pip install --upgrade pip ipython requests
48
# Install some packages in specific variants
49
-pip install torch --index-url https://download.pytorch.org/whl/cpu
49
+pip install torch==2.7.0+cpu torchvision==0.22.0+cpu --index-url https://download.pytorch.org/whl/cpu
50
51
52
echo "====================PYTHON END===================="
requirements.txt
+5
-2
@@ -29,8 +29,11 @@ playwright==1.52.0
29
pypdf==4.3.1
30
python-dotenv==1.1.0
31
pytz==2024.2
32
-sentence-transformers==3.0.1
33
-tiktoken==0.8.0
32
+sentence-transformers[cpu]==4.1.0
33
+tiktoken==0.9.0
34
+torch[cpu]==2.7.0+cpu
35
+torchvision[cpu]==0.22.0+cpu
36
+transformers[cpu]==4.52.4
37
unstructured==0.15.13
38
unstructured-client==0.25.9
39
webcolors==24.6.0