19 lines
401 B
Plaintext
19 lines
401 B
Plaintext
# Install the CUDA build of torch that matches your machine if the default wheel
|
|
# is not appropriate for your environment.
|
|
torch==2.9.1
|
|
transformers==5.10.2
|
|
numpy==2.4.4
|
|
PyYAML==6.0.3
|
|
tqdm==4.67.3
|
|
tensorboard==2.20.0
|
|
matplotlib==3.10.9
|
|
triton==3.5.1
|
|
typing_extensions==4.15.0
|
|
sentencepiece==0.2.1
|
|
safetensors==0.7.0
|
|
prettytable==3.17.0
|
|
|
|
# Data preparation dependencies.
|
|
datasets==4.8.5
|
|
openai==2.6.1
|