forked from instructlab/instructlab
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathrequirements.txt
More file actions
44 lines (43 loc) · 1 KB
/
requirements.txt
File metadata and controls
44 lines (43 loc) · 1 KB
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
44
# SPDX-License-Identifier: Apache-2.0
boto3>=1.35.96
click>=8.1.7
click-didyoumean>=0.3.0
datasets>=2.18.0
filelock
gguf>=0.6.0
GitPython>=3.1.42
httpx>=0.25.0
instructlab-eval>=0.5.1
instructlab-quantize>=0.1.0
instructlab-schema>=0.4.2
instructlab-sdg>=0.7.3
instructlab-training>=0.10.1
llama_cpp_python[server]>=0.3.8
mlx>=0.5.1,<0.6.0; sys_platform == 'darwin' and platform_machine == 'arm64'
numpy>=1.26.4
openai>=1.13.3
peft>=0.9.0
prompt-toolkit>=3.0.38
pydantic>=2.7.4
pydantic_yaml>=1.2.0
PyYAML>=6.0.0
rich>=13.3.1
rouge-score>=0.1.2
ruamel.yaml>=0.17.0
sentencepiece>=0.2.0
# "old" version required for vLLM on CUDA to build
tokenizers>=0.11.1
toml>=0.10.2
# Default version. Can be overridden in extra requirements. Do not cap this dependency here. Cap in constraints-dev.txt
torch>=2.6.0
tqdm>=4.66.2
transformers>=4.41.2
trl>=0.12.2
wandb>=0.16.4
xdg-base-dirs>=6.0.1
psutil>=6.0.0
huggingface_hub[hf_transfer]>=0.1.8
haystack-ai>=2.8
docling-core[chunking]>=2.10.0
docling>=2.18.0
sentence-transformers>=3.0.0