From 59566c54f4fb814dd94787d348f18635563e90a3 Mon Sep 17 00:00:00 2001 From: David Chatel Date: Wed, 8 Oct 2025 13:48:42 +0200 Subject: [PATCH] fixed slow landmarks onnx --- utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/utils.py b/utils.py index f8f9fe7..3bb65f8 100644 --- a/utils.py +++ b/utils.py @@ -8,7 +8,7 @@ from scipy.spatial import ConvexHull from folder_paths import models_dir from .BiSeNet import BiSeNet from ultralytics import YOLO -from onnxruntime import InferenceSession, get_available_providers +from onnxruntime import InferenceSession from transformers import SegformerImageProcessor, SegformerForSemanticSegmentation from skimage import transform as trans @@ -62,7 +62,7 @@ class Models: @classmethod def lmk(cls, crop): if '_lmk' not in cls.__dict__: - cls._lmk = InferenceSession(os.path.join(models_dir, 'landmarks', 'fan2_68_landmark.onnx')) + cls._lmk = InferenceSession(os.path.join(models_dir, 'landmarks', 'fan2_68_landmark.onnx'), providers=['CUDAExecutionProvider', 'CPUExecutionProvider']) lmk = cls._lmk.run(None, {'input': crop})[0] return lmk