einops
numpy
pytorch_ignite
torch
torchaudio

[all]
dasheng[train]

[train]
accelerate>=0.28.0
bitsandbytes>=0.35.4
webdataset>=0.2.86
braceexpand>=0.1.7
fire>=0.5.0
loguru>=0.7.2
numpy>=1.24.1
pytorch_ignite>=0.4.13
PyYAML>=6.0.1
torch>=2.1.1
torchaudio>=2.1.1
tqdm>=4.66.1
pandas>=2.0
