fix: torchvision version to avoid compatibility issue (#866)

* fix: fix torchvision version to avoid compatibility issue * fix: fix torchvision version to avoid compatibility issue * fix: fix python version for torchvision * fix: fix python version for torchvision
jina-ai · Nov 25, 2022 · 67f551c · 67f551c
1 parent 0223e6f
commit 67f551c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/server/setup.py b/server/setup.py
@@ -45,7 +45,7 @@
         'ftfy',
         'torch',
         'regex',
-        'torchvision',
+        'torchvision<=0.13.0' if sys.version_info <= (3, 7, 2) else 'torchvision',
         'jina>=3.8.0',
         'prometheus-client',
         'open_clip_torch>=1.3.0',