diff --git a/requirements.txt b/requirements.txt index f36b280..8fe00aa 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ pydantic -torchvision==0.16 +torchvision==0.16.0 pytorch-lightning==2.1.3 torch[cpu]==2.1.0 PVNet-summation==0.1.2