Metadata-Version: 2.4
Name: cosmos-predict2
Version: 1.0.9
Summary: Cosmos World Foundation Model
Project-URL: documentation, https://github.com/nvidia-cosmos/cosmos-predict2/blob/main/README.md
Project-URL: homepage, https://research.nvidia.com/labs/dir/cosmos-predict2
Project-URL: issues, https://github.com/nvidia-cosmos/cosmos-predict2/issues
Project-URL: repository, https://github.com/nvidia-cosmos/cosmos-predict2
Author: NVIDIA Corporation
License: Apache-2.0
License-File: LICENSE
Classifier: Development Status :: 4 - Beta
Classifier: Environment :: GPU :: NVIDIA CUDA
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Operating System :: POSIX :: Linux
Classifier: Programming Language :: Python
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Python: >=3.10
Requires-Dist: attrs==25.1.0
Requires-Dist: better-profanity==0.7.0
Requires-Dist: boto3==1.35.99
Requires-Dist: decord==0.6.0
Requires-Dist: diffusers==0.33.1
Requires-Dist: einops==0.8.1
Requires-Dist: ftfy==6.3.1
Requires-Dist: fvcore==0.1.5.post20221221
Requires-Dist: huggingface-hub==0.32.4
Requires-Dist: hydra-core==1.3.2
Requires-Dist: imageio[ffmpeg,pyav]==2.37.0
Requires-Dist: iopath==0.1.10
Requires-Dist: ipdb==0.13.13
Requires-Dist: loguru==0.7.2
Requires-Dist: mediapy==1.2.2
Requires-Dist: megatron-core==0.10.0
Requires-Dist: modelscope==1.26.0
Requires-Dist: nltk==3.9.1
Requires-Dist: numpy==1.26.4
Requires-Dist: nvidia-ml-py==12.535.133
Requires-Dist: omegaconf==2.3.0
Requires-Dist: opencv-python==4.10.0.84
Requires-Dist: pandas==2.2.3
Requires-Dist: peft==0.14.0
Requires-Dist: pillow==11.1.0
Requires-Dist: protobuf==4.25.3
Requires-Dist: pycocotools==2.0.10
Requires-Dist: pynvml==12.0.0
Requires-Dist: pyyaml==6.0.2
Requires-Dist: qwen-vl-utils[decord]==0.0.11
Requires-Dist: retinaface-py==0.0.2
Requires-Dist: safetensors==0.5.3
Requires-Dist: scikit-image==0.25.2
Requires-Dist: sentencepiece==0.2.0
Requires-Dist: setuptools
Requires-Dist: termcolor==2.5.0
Requires-Dist: tqdm==4.66.5
Requires-Dist: transformers==4.51.3
Requires-Dist: triton==3.2.0
Requires-Dist: webdataset==0.2.111
Provides-Extra: cu126
Requires-Dist: flash-attn==2.6.3; extra == 'cu126'
Requires-Dist: natten==0.21.0; extra == 'cu126'
Requires-Dist: nvidia-cublas-cu12==12.6.4.1; extra == 'cu126'
Requires-Dist: nvidia-cuda-cupti-cu12==12.6.80; extra == 'cu126'
Requires-Dist: nvidia-cuda-nvrtc-cu12==12.6.77; extra == 'cu126'
Requires-Dist: nvidia-cuda-runtime-cu12==12.6.77; extra == 'cu126'
Requires-Dist: nvidia-cudnn-cu12==9.5.1.17; extra == 'cu126'
Requires-Dist: nvidia-cufft-cu12==11.3.0.4; extra == 'cu126'
Requires-Dist: nvidia-curand-cu12==10.3.7.77; extra == 'cu126'
Requires-Dist: nvidia-cusolver-cu12==11.7.1.2; extra == 'cu126'
Requires-Dist: nvidia-cusparse-cu12==12.5.4.2; extra == 'cu126'
Requires-Dist: nvidia-cusparselt-cu12==0.6.3; extra == 'cu126'
Requires-Dist: nvidia-nccl-cu12==2.21.5; extra == 'cu126'
Requires-Dist: nvidia-nvjitlink-cu12==12.6.85; extra == 'cu126'
Requires-Dist: nvidia-nvtx-cu12==12.6.77; extra == 'cu126'
Requires-Dist: torch==2.6.0; extra == 'cu126'
Requires-Dist: torchvision==0.21.0; extra == 'cu126'
Requires-Dist: transformer-engine==1.13; extra == 'cu126'
Description-Content-Type: text/markdown

# Cosmos-Predict2

[Documentation](https://github.com/nvidia-cosmos/cosmos-predict2/blob/main/README.md)
