transformers==4.45.2
huggingface-hub==0.28.1
requests
certifi
tqdm
numpy==1.25.2
torch==2.4.1
torchvision==0.19.1
torchaudio==2.4.1
accelerate==1.3.0
sapiens-machine
sapiens-accelerator
sapiens-generalization
tokenizers==0.20.1
regex
datasets
sentencepiece
protobuf
optimum
einops
nemo-toolkit==2.1.0
hydra-core
lightning
braceexpand
webdataset
h5py
ijson
matplotlib
diffusers==0.32.2
moviepy==1.0.3
llama-cpp-python==0.3.6
llamacpp==0.1.14
beautifulsoup4
av
ftfy
tiktoken
opencv-python
scipy
TTS==0.22.0
pydub
megatron-core==0.10.0

[multimedia]

[multimedia:python_version < "3.12"]
pyav
TTS==0.22.0

[multimedia:python_version >= "3.12"]
av

[toolkit]
nemo-toolkit[all]==2.1.0
