From 0be974972b5719e3670269e7a15bf5f2e329b3b3 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Tue, 19 May 2026 09:09:02 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- studio/backend/core/training/worker.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/studio/backend/core/training/worker.py b/studio/backend/core/training/worker.py index cea3cd480c..aa98a20366 100644 --- a/studio/backend/core/training/worker.py +++ b/studio/backend/core/training/worker.py @@ -81,6 +81,7 @@ _WINDOWS_ROCM_GROUPED_MM_LIB = None # scope so they are not garbage collected. _ROCM_DLL_HANDLES: list = [] if sys.platform == "win32": + def _add_rocm_dll_dirs_worker() -> None: _candidates: list[str] = [] for _var in ("HIP_PATH", "ROCM_PATH"): @@ -104,6 +105,7 @@ if sys.platform == "win32": _ROCM_DLL_HANDLES.append(os.add_dll_directory(_d)) except (OSError, AttributeError): pass + _add_rocm_dll_dirs_worker() del _add_rocm_dll_dirs_worker @@ -1968,6 +1970,7 @@ def run_training_process( if sys.platform == "win32": try: import torch as _torch_probe + _is_win32_rocm = bool( getattr(getattr(_torch_probe, "version", None), "hip", None) or "rocm" in getattr(_torch_probe, "__version__", "").lower()