Fix PyTorch version check for nightly builds (openai#173)

ljl02521 · Nov 5, 2021 · 1a8b4b2 · 1a8b4b2
1 parent 2867559
commit 1a8b4b2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/clip/clip.py b/clip/clip.py
@@ -19,7 +19,7 @@
     BICUBIC = Image.BICUBIC
 
 
-if [int(n.split("+")[0]) for n in torch.__version__.split(".")] < [1, 7, 1]:
+if [int(n.split("+")[0]) for n in torch.__version__.split(".")[:3]] < [1, 7, 1]:
     warnings.warn("PyTorch version 1.7.1 or higher is recommended")