From 827ce1e9dfba14cc63ce9b3b193a5fab09a4e682 Mon Sep 17 00:00:00 2001
From: SameetAsadullah <sameetassadullah744@gmail.com>
Date: Thu, 19 Oct 2023 09:50:12 +0000
Subject: [PATCH] Fix torch version checking

---
 facelib/detection/yolov5face/face_detector.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/facelib/detection/yolov5face/face_detector.py b/facelib/detection/yolov5face/face_detector.py
index 3d54587..ff486be 100644
--- a/facelib/detection/yolov5face/face_detector.py
+++ b/facelib/detection/yolov5face/face_detector.py
@@ -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):