torch<3,>=2.1
numpy<3,>=1.24
sentencepiece<0.3,>=0.2
transformers<6,>=4.51

[all]
torch<3,>=2.1
torchvision<1,>=0.16
torchaudio<3,>=2.1
flask<4,>=3
beautifulsoup4<5,>=4.12
PyYAML<7,>=6
tqdm<5,>=4.66
accelerate<2,>=0.30
datasets<4,>=2.18
peft<1,>=0.11
safetensors<1,>=0.4

[all-cpu]
torch<3,>=2.1
torchvision<1,>=0.16
torchaudio<3,>=2.1
flask<4,>=3
beautifulsoup4<5,>=4.12
PyYAML<7,>=6
tqdm<5,>=4.66
accelerate<2,>=0.30
datasets<4,>=2.18
peft<1,>=0.11
safetensors<1,>=0.4

[all-cuda]
torch<3,>=2.1
torchvision<1,>=0.16
torchaudio<3,>=2.1
flask<4,>=3
beautifulsoup4<5,>=4.12
PyYAML<7,>=6
tqdm<5,>=4.66
accelerate<2,>=0.30
datasets<4,>=2.18
peft<1,>=0.11
safetensors<1,>=0.4

[all-cuda121]
torch<3,>=2.1
torchvision<1,>=0.16
torchaudio<3,>=2.1
flask<4,>=3
beautifulsoup4<5,>=4.12
PyYAML<7,>=6
tqdm<5,>=4.66
accelerate<2,>=0.30
datasets<4,>=2.18
peft<1,>=0.11
safetensors<1,>=0.4

[cpu]
torch<3,>=2.1
torchvision<1,>=0.16
torchaudio<3,>=2.1

[cuda]
torch<3,>=2.1
torchvision<1,>=0.16
torchaudio<3,>=2.1

[cuda121]
torch<3,>=2.1
torchvision<1,>=0.16
torchaudio<3,>=2.1

[dev]
pytest>=8
build<2,>=1.2
twine<7,>=5

[examples]
accelerate<2,>=0.30
datasets<4,>=2.18
peft<1,>=0.11
safetensors<1,>=0.4

[hf]
accelerate<2,>=0.30
datasets<4,>=2.18
peft<1,>=0.11

[peft]
peft<1,>=0.11

[preprocess]
beautifulsoup4<5,>=4.12
PyYAML<7,>=6
tqdm<5,>=4.66

[web]
flask<4,>=3
