diff --git a/requirements/extra.txt b/requirements/extra.txt index 0e7dffbcb39b0..85437327bce06 100644 --- a/requirements/extra.txt +++ b/requirements/extra.txt @@ -3,7 +3,7 @@ matplotlib>3.1 horovod>=0.21.2 # no need to install with [pytorch] as pytorch is already installed omegaconf>=2.0.1 -torchtext>=0.5, <0.7 # TODO: temporary fix fix for compatibility +torchtext>=0.5 onnx>=1.7.0 onnxruntime>=1.3.0 hydra-core>=1.0