plamo-2-1b / requirements.txt
masanorihirano's picture
Update requirements.txt
6cdcba1 verified
raw
history blame contribute delete
206 Bytes
huggingface_hub==0.25.2
--extra-index-url https://download.pytorch.org/whl/cu124
torch==2.5.1+cu124
numpy>=1.26.4,<2.0
numba>=0.60.0
transformers>=4.44.2
mamba_ssm>=2.2.2
causal_conv1d>=1.4.0
gradio==5.0.1