PyPI Stats

Search

All packages
Top packages

Track packages

cosmos-predict2


PyPI page
Home page
Author: NVIDIA Corporation
License: Apache-2.0
Summary: Cosmos World Foundation Model
Latest version: 1.0.9
Required dependencies: attrs | better-profanity | boto3 | decord | diffusers | einops | ftfy | fvcore | huggingface-hub | hydra-core | imageio | iopath | ipdb | loguru | mediapy | megatron-core | modelscope | nltk | numpy | nvidia-ml-py | omegaconf | opencv-python | pandas | peft | pillow | protobuf | pycocotools | pynvml | pyyaml | qwen-vl-utils | retinaface-py | safetensors | scikit-image | sentencepiece | setuptools | termcolor | tqdm | transformers | triton | webdataset
Optional dependencies: flash-attn | natten | nvidia-cublas-cu12 | nvidia-cuda-cupti-cu12 | nvidia-cuda-nvrtc-cu12 | nvidia-cuda-runtime-cu12 | nvidia-cudnn-cu12 | nvidia-cufft-cu12 | nvidia-curand-cu12 | nvidia-cusolver-cu12 | nvidia-cusparse-cu12 | nvidia-cusparselt-cu12 | nvidia-nccl-cu12 | nvidia-nvjitlink-cu12 | nvidia-nvtx-cu12 | torch | torchvision | transformer-engine

Downloads last day: 44
Downloads last week: 169
Downloads last month: 899