elungky commited on
Commit
17c6444
·
1 Parent(s): 8bf5f16

Add 'attrs' package to cosmos-predict1.yaml dependencies

Browse files
Files changed (1) hide show
  1. cosmos-predict1.yaml +1 -0
cosmos-predict1.yaml CHANGED
@@ -30,4 +30,5 @@ dependencies:
30
  - IPython
31
  - megatron.core
32
  - attrs
 
33
  # PyTorch and TorchVision will be installed separately in the Dockerfile
 
30
  - IPython
31
  - megatron.core
32
  - attrs
33
+ - iopath
34
  # PyTorch and TorchVision will be installed separately in the Dockerfile