From 2867559c5fe0b02a2d3167aeacd333b3c4276847 Mon Sep 17 00:00:00 2001 From: Santiago Castro Date: Thu, 4 Nov 2021 17:32:10 -0400 Subject: [PATCH] Fix PyTorch version check (#160) * Fix PyTorch version check * Fix suggestion Co-authored-by: Jong Wook Kim --- clip/clip.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/clip/clip.py b/clip/clip.py index 9d6f4d6..3c8433c 100644 --- a/clip/clip.py +++ b/clip/clip.py @@ -19,7 +19,7 @@ except ImportError: BICUBIC = Image.BICUBIC -if torch.__version__.split(".") < ["1", "7", "1"]: +if [int(n.split("+")[0]) for n in torch.__version__.split(".")] < [1, 7, 1]: warnings.warn("PyTorch version 1.7.1 or higher is recommended")