forked from HandsOnLLM/Hands-On-Large-Language-Models
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathrequirements.txt
43 lines (39 loc) · 898 Bytes
/
requirements.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
# Data handling
numpy == 1.26.4
pandas == 2.2.2
datasets == 2.20.0
# Environment
jupyterlab == 4.2.2
ipywidgets == 8.1.3
# Hard dependencies throughout the entire book
torch == 2.3.1
transformers == 4.41.2
sentence-transformers == 3.0.1
matplotlib == 3.9.0
scikit-learn == 1.5.0
sentencepiece == 0.2.0
nltk == 3.8.1
evaluate == 0.4.2
scipy == 1.12.0 # TEMPORARY DEP BECAUSE OF https://github.com/piskvorky/gensim/issues/3525
# Cloud providers
openai == 1.34.0
cohere == 5.5.8
# Chapter-specific dependencies
datamapplot == 0.3.0
faiss-cpu==1.8.0
bertopic == 0.16.3
annoy == 1.17.3
llama_cpp_python == 0.2.78 -C cmake.args="-DLLAMA_BLAS=ON"
numexpr == 2.10.0
langchain == 0.2.5
langchain-community == 0.2.5
langchain-openai==0.1.8
duckduckgo-search == 6.1.6
gensim == 4.3.2
setfit == 1.0.3
seqeval == 1.2.2
trl == 0.9.4
peft == 0.11.1
accelerate == 0.31.0
bitsandbytes == 0.43.1
mteb == 1.12.39