diff options
Diffstat (limited to 'trm/specific_requirements.txt')
| -rw-r--r-- | trm/specific_requirements.txt | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/trm/specific_requirements.txt b/trm/specific_requirements.txt new file mode 100644 index 0000000..a9ea806 --- /dev/null +++ b/trm/specific_requirements.txt @@ -0,0 +1,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 |