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

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
@@ -29,4 +29,5 @@ dependencies:
29
  - megatron
30
  - IPython
31
  - megatron.core
 
32
  # PyTorch and TorchVision will be installed separately in the Dockerfile
 
29
  - megatron
30
  - IPython
31
  - megatron.core
32
+ - attrs
33
  # PyTorch and TorchVision will be installed separately in the Dockerfile