veerukhannan commited on
Commit
41dc1a2
·
verified ·
1 Parent(s): a8dcc53

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +9 -6
requirements.txt CHANGED
@@ -11,9 +11,10 @@ torch>=2.1.0
11
  accelerate>=0.25.0
12
  sentence-transformers>=2.2.2
13
 
14
- # Database and utilities
15
- astrapy==0.7.0
16
- cassio>=0.1.0
 
17
  python-dotenv==1.0.0
18
  huggingface-hub>=0.19.4
19
 
@@ -23,6 +24,8 @@ pandas>=1.3.0
23
  tqdm>=4.65.0
24
  pytest>=7.0.0
25
 
26
- # Optional: Performance improvements
27
- bitsandbytes>=0.41.0 # For 8-bit quantization
28
- safetensors>=0.4.0 # For faster model loading
 
 
 
11
  accelerate>=0.25.0
12
  sentence-transformers>=2.2.2
13
 
14
+ # Vector Database
15
+ chromadb>=0.4.15
16
+
17
+ # Utilities
18
  python-dotenv==1.0.0
19
  huggingface-hub>=0.19.4
20
 
 
24
  tqdm>=4.65.0
25
  pytest>=7.0.0
26
 
27
+ # Performance improvements
28
+ bitsandbytes>=0.41.0
29
+ safetensors>=0.4.0
30
+ einops>=0.6.1
31
+ tiktoken>=0.5.1