Starrick Liu
dce38a7556
* Update Wall-X dependency and runtime setup * Simplify FlashAttention installation notes
27 lines
480 B
Plaintext
27 lines
480 B
Plaintext
torch==2.10.0
|
|
torchvision==0.25.0
|
|
torchaudio==2.10.0
|
|
transformers==5.2.0
|
|
accelerate==1.13.0
|
|
peft==0.18.1
|
|
numpy>=1.26,<2
|
|
numba>=0.60,<0.62
|
|
datasets>=2.20,<5
|
|
pyarrow>=14,<23
|
|
opencv-python-headless>=4.11,<4.12
|
|
Pillow>=10,<12
|
|
packaging>=24,<26
|
|
ninja>=1.11,<2
|
|
scipy==1.15.3
|
|
pyyaml>=6.0
|
|
safetensors>=0.8,<0.9
|
|
torchdiffeq==0.2.5
|
|
qwen_vl_utils==0.0.11
|
|
websockets>=12.0,<16
|
|
msgpack-numpy>=0.4.8,<0.5
|
|
numpydantic>=1.8,<2
|
|
tqdm>=4.67,<5
|
|
tyro>=0.9.0,<1.0
|
|
wandb>=0.23,<0.25
|
|
matplotlib>=3.10,<4
|