diff --git a/requirements.txt b/requirements.txt index 425c9ac..52731b6 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ -torch>=1.9.1 +torch>=1.9.1,<2.6.0 torchvision>=0.10.1 Pillow>=7.1.2 PyYAML>=5.3.1