Fix torch version checking

2023-10-19 09:50:12 +00:00
parent 45ebbe5eb0
commit 827ce1e9df
1 changed files with 3 additions and 1 deletions
@@ -17,7 +17,9 @@ from custom_nodes.facerestore_cf.facelib.detection.yolov5face.utils.general impo
    scale_coords_landmarks,
 )

-IS_HIGH_VERSION = True
+from torch import torch_version
+
+IS_HIGH_VERSION = torch_version.__version__ >= "1.9.0"


 def isListempty(inList):