gen3c / cosmos-predict1.yaml
elungky's picture
Add 'einops' package to cosmos-predict1.yaml dependencies
3cc0f0e
raw
history blame
424 Bytes
name: cosmos-predict1
channels:
- pytorch
- nvidia
- conda-forge
- defaults
dependencies:
- python=3.10
- pip=25.0
- cmake
- ninja
- gcc=12.4.0
- gxx=12.4.0
- pytorch=2.3.1
- torchvision=0.18.1
- torchaudio=2.3.1
- pytorch-cuda=12.1
- pip:
- -r file:///app/gui/requirements.txt
- uvicorn[standard]
- fastapi
- git+https://github.com/microsoft/MoGe.git
- einops