Spaces:
Build error
Build error
Add 'attrs' package to cosmos-predict1.yaml dependencies
Browse files- cosmos-predict1.yaml +1 -0
cosmos-predict1.yaml
CHANGED
|
@@ -31,4 +31,5 @@ dependencies:
|
|
| 31 |
- megatron.core
|
| 32 |
- attrs
|
| 33 |
- iopath
|
|
|
|
| 34 |
# PyTorch and TorchVision will be installed separately in the Dockerfile
|
|
|
|
| 31 |
- megatron.core
|
| 32 |
- attrs
|
| 33 |
- iopath
|
| 34 |
+
- transformer_engine
|
| 35 |
# PyTorch and TorchVision will be installed separately in the Dockerfile
|