diff options
| author | YurenHao0426 <blackhao0426@gmail.com> | 2026-06-13 12:35:36 -0500 |
|---|---|---|
| committer | YurenHao0426 <blackhao0426@gmail.com> | 2026-06-13 12:35:36 -0500 |
| commit | 66e0d8b9fd4d0f7a2231d689c055e26fdf1cf04a (patch) | |
| tree | c29cba61124018755a19b02c9d33e3ad5f2e05cc /env/pip-freeze.txt | |
Curated export for clone-and-run Maze training (2x A6000) + diagnostics.
trm/hrm pretrain.py carry trajectory-augmentation code (backward-compatible).
Heavy artifacts (checkpoints/wandb/npz) gitignored; see PROVENANCE.md.
Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
Diffstat (limited to 'env/pip-freeze.txt')
| -rw-r--r-- | env/pip-freeze.txt | 18 |
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 |