Merge pull request #150 from hben35096/main

Reorder execution providers in model initialization
2026-09-18 10:07:42 +08:00
parent 7b678b401a 5f68ae0de7
commit 51ee6a999b
1 changed files with 1 additions and 1 deletions
@@ -73,7 +73,7 @@ class LS_HumanPartsUltra:
        """
        import onnxruntime as ort

-        model = ort.InferenceSession(model_path, providers=['TensorrtExecutionProvider', 'CUDAExecutionProvider', 'CPUExecutionProvider'])
+        model = ort.InferenceSession(model_path, providers=['CUDAExecutionProvider', 'CPUExecutionProvider', 'TensorrtExecutionProvider'])
        ret_images = []
        ret_masks = []
        for img in image: