diff --git a/requirements.txt b/requirements.txt index 9babb4f9b1..8aaa1ec97d 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,6 @@ torch>=1.7 # TODO: regenerate weights with lower PT version torchmetrics -torchvision>=0.8 # TODO: lower to 0.7 after PT 1.6 +torchvision==0.8 # TODO: lower to 0.7 after PT 1.6 pytorch-lightning>=1.3.0rc1 lightning-bolts>=0.3.3 PyYAML>=5.1 @@ -16,6 +16,6 @@ rouge-score>=0.0.4 sentencepiece>=0.1.95 filelock # comes with 3rd-party dependency pycocotools>=2.0.2 ; python_version >= "3.7" -kornia>=0.5.0 +kornia==0.5.0 pytorchvideo matplotlib # used by the visualisation callback