AlixJabda commited on
Commit
5983a6d
·
1 Parent(s): d1ca049

pin torch>=2.4 and transformers 5.x; numpy<2

Browse files
Files changed (1) hide show
  1. requirements.txt +12 -6
requirements.txt CHANGED
@@ -1,14 +1,20 @@
1
  --extra-index-url https://download.pytorch.org/whl/cpu
2
 
3
- torch==2.1.2+cpu
4
- torchvision==0.16.2+cpu
5
 
 
 
 
 
 
 
 
6
  fastapi
7
  uvicorn[standard]
8
- chromadb
9
- sentence-transformers
10
  groq
11
  python-dotenv
12
  pydantic
13
- huggingface_hub
14
- chromadb==1.5.9
 
1
  --extra-index-url https://download.pytorch.org/whl/cpu
2
 
3
+ # NumPy 2.x kann mit einigen Torch-Builds Probleme machen -> stabil pinnen
4
+ numpy==1.26.4
5
 
6
+ # Transformers 5.x braucht PyTorch >= 2.4
7
+ torch==2.4.1+cpu
8
+
9
+ # Transformers auf 5.x fixieren
10
+ transformers==5.9.0
11
+
12
+ # Deine App-Dependencies
13
  fastapi
14
  uvicorn[standard]
15
+ chromadb==1.5.9
16
+ sentence-transformers==5.5.1
17
  groq
18
  python-dotenv
19
  pydantic
20
+ huggingface_hub