Cosmos-Reason1-Demo / requirements.txt
harrim-nv's picture
Update requirements.txt
5487fe3 verified
raw
history blame contribute delete
390 Bytes
cmake
fastapi
uvicorn
transformers>=4.36.0
datasets
pydantic
math_verify
pybind11[global]
vllm>=0.3.0
torch==2.5.1
torchvision
tensordict
toml
tomli===2.2.1
StrEnum
redis
msgpack
qwen-vl-utils==0.0.11
# flash-attn==2.3.6
setuptools==79.0.1
pynvml
torchao
gradio>=4.0.0
ninja
packaging
accelerate
safetensors
numpy<2.0.0
pandas>=2.0.0,<2.2.3dev0
scikit-learn<2.0.0
pyarrow>=16.1.0,<16.2.0a0