summaryrefslogtreecommitdiff
path: root/env/pip-freeze.txt
diff options
context:
space:
mode:
Diffstat (limited to 'env/pip-freeze.txt')
-rw-r--r--env/pip-freeze.txt18
1 files changed, 18 insertions, 0 deletions
diff --git a/env/pip-freeze.txt b/env/pip-freeze.txt
new file mode 100644
index 0000000..03c81b2
--- /dev/null
+++ b/env/pip-freeze.txt
@@ -0,0 +1,18 @@
+adam_atan2==0.0.3
+argdantic==1.3.3
+coolname==2.2.0
+einops==0.8.1
+flash_attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.8.3/flash_attn-2.8.3+cu12torch2.7cxx11abiTRUE-cp310-cp310-linux_x86_64.whl#sha256=ce91e246f21d61ad66b1a7555340dbaa28e4aa86edcf00c18f0837422939b529
+huggingface-hub==0.34.4
+hydra-core==1.3.2
+numba==0.61.2
+omegaconf==2.3.0
+pydantic==2.11.7
+pydantic_core==2.33.2
+pydantic-settings==2.14.1
+torch==2.7.0+cu126
+torchaudio==2.7.0+cu126
+torchvision==0.22.0+cu126
+tqdm==4.67.1
+triton==3.3.0
+wandb==0.21.3