Use webgpu for local development

Flova · Flova · commit c38973c0d52b · 2025-12-28T12:47:43.000+01:00
Signed-off-by: Florian Vahl &lt;florian@flova.de&gt;
diff --git a/pixi.lock b/pixi.lock
diff --git a/pixi.toml b/pixi.toml
@@ -193,7 +193,7 @@ syrupy = ">=5.0.0, <6"
 exhale = ">=0.3.7, <0.4"
 mycroft-mimic3-tts = ">=0.2.4, <0.3"
 pyttsx3 = ">=2.99, <3"
-onnxruntime-gpu = ">=1.23.2, <2"
+onnxruntime-webgpu = ">=1.24.0.dev20251218001, <2"
 
 [feature.robot.pypi-dependencies]
 # These are only needed on the robot
diff --git a/src/bitbots_vision/bitbots_vision/vision_modules/yoeo/yoeo_handlers.py b/src/bitbots_vision/bitbots_vision/vision_modules/yoeo/yoeo_handlers.py
@@ -229,7 +229,7 @@ def __init__(
             raise ImportError("Could not import onnxruntime. The selected handler requires this package.") from e
 
         logger.debug(f"Loading file...\n\t{onnx_path}")
-        self._inference_session = onnxruntime.InferenceSession(onnx_path)
+        self._inference_session = onnxruntime.InferenceSession(onnx_path, providers=["WebGpuExecutionProvider"])
         self._input_layer = self._inference_session.get_inputs()[0]
 
         self._img_preprocessor: utils.IImagePreProcessor = utils.DefaultImagePreProcessor(
