# Exact dependency versions the bind2_0 checkpoints were trained/exported with.
# Source: local conda env `babylm-smoke` (C:/Users/yulin/.conda/envs/babylm-smoke),
# queried via pip on 2026-07-15.
#
# Python: 3.11.15
#
# NOTE on fla (flash-linear-attention): the inlined HF modeling code
# (modeling_babylm.py) DOES import it — lazily, inside GDNBlock:
#   `from fla.layers import GatedDeltaNet`  (only executed when arch == "bind2_0")
# Since the bind2_0 exports instantiate GDN blocks, fla IS required at runtime
# to load/run these checkpoints. Installed from PyPI as release 0.5.1
# (no commit-pin / direct-URL metadata present in the env; pulls fla-core 0.5.1).
#
# NOTE on triton: the env uses the Windows fork `triton-windows`
# (github.com/woct0rdho/triton-windows); on Linux use the matching upstream
# `triton` that your torch build requires.
# torch build is CUDA 12.6 (`+cu126`); pick the equivalent build for your platform.

torch==2.12.1+cu126
transformers==5.13.0
triton-windows==3.7.1.post27
flash-linear-attention==0.5.1
fla-core==0.5.1
safetensors==0.8.0
numpy==2.4.6