trm/requirements.txt


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

torch
adam-atan2
einops
tqdm
coolname
pydantic
argdantic
wandb
omegaconf
hydra-core
huggingface_hub
packaging 
ninja 
wheel 
setuptools 
setuptools-scm
pydantic-core
huggingface_hub 
numba
triton