torch~=2.5.1
torchvision~=0.20.1
torchaudio~=2.5.1
tqdm~=4.67.0
transformers~=4.46.2
einops~=0.8.0
tiktoken~=0.8.0