pin torch==2.1.0, torchvision=0.16.0

This commit is contained in:
psychedelicious 2023-11-16 09:28:29 +11:00
parent b6f356f067
commit 36b8549f3a

View File

@ -79,8 +79,8 @@ dependencies = [
"semver~=3.0.1",
"send2trash",
"test-tube~=0.7.5",
"torch~=2.1.0",
"torchvision~=0.16",
"torch==2.1.0",
"torchvision==0.16.0",
"torchmetrics~=0.11.0",
"torchsde~=0.2.5",
"transformers~=4.35.0",