Cosmos-Reason1-Demo / requirements.txt
harrim-nv's picture
Update requirements.txt
59f0fba verified
raw
history blame
403 Bytes
cmake
fastapi
uvicorn
transformers>=4.36.0
datasets
pydantic
math_verify
pybind11[global]
vllm>=0.3.0
torch==2.6.0
torchvision==0.21.0
tensordict==0.7.2
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