Small updates.

2024-12-19 13:52:44 +01:00
parent 19bb1a6372
commit 2ad20549f2
2 changed files with 8 additions and 10 deletions
--- a/scripts/utils_2d_pose_ort.py
+++ b/scripts/utils_2d_pose_ort.py
@ -49,9 +49,10 @@ class BaseModel(ABC):
                    },
                )
            )
-        if usegpu and "CUDAExecutionProvider" in providers:
+        elif usegpu and "CUDAExecutionProvider" in providers:
            self.providers.append("CUDAExecutionProvider")
-        self.providers.append("CPUExecutionProvider")
+        else:
+            self.providers.append("CPUExecutionProvider")
        print("Using providers:", self.providers)

        self.session = ort.InferenceSession(