update

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