question

immich-app · Jul 12, 2024 · d706380 · d706380
1 parent c90b2b5
commit d706380
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/machine-learning/app/sessions/ort.py b/machine-learning/app/sessions/ort.py
@@ -86,6 +86,7 @@ def _provider_options_default(self) -> list[dict[str, Any]]:
         options = []
         for provider in self.providers:
             match provider:
+                # Should we add something for ROCMExecutionProvider?
                 case "CPUExecutionProvider" | "CUDAExecutionProvider":
                     option = {"arena_extend_strategy": "kSameAsRequested"}
                 case "OpenVINOExecutionProvider":