trm/specific_requirements.txt


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

torch==2.7.0+cu126
adam-atan2==0.0.3
einops==0.8.1
tqdm==4.67.1
coolname==2.2.0
pydantic==2.11.7
argdantic==1.3.3
wandb==0.21.3
omegaconf==2.3.0
hydra-core==1.3.2
huggingface_hub==0.34.4
packaging==24.1
ninja==1.13.0
wheel==0.45.1
setuptools==80.9.0
setuptools-scm==9.2.0
pydantic-core==2.33.2
huggingface_hub==0.34.4
numba==0.61.2
triton==3.3.0