From 56c95c69fbca1352b1ee23e77befcada1776f980 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Fri, 24 Apr 2026 15:50:40 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- studio/backend/core/inference/llama_cpp.py | 1 + 1 file changed, 1 insertion(+) diff --git a/studio/backend/core/inference/llama_cpp.py b/studio/backend/core/inference/llama_cpp.py index 82ca3f807b..388bfee837 100644 --- a/studio/backend/core/inference/llama_cpp.py +++ b/studio/backend/core/inference/llama_cpp.py @@ -1805,6 +1805,7 @@ class LlamaCppBackend: env["CUDA_VISIBLE_DEVICES"] = pinned try: import torch as _torch + if getattr(_torch.version, "hip", None) is not None: env["HIP_VISIBLE_DEVICES"] = pinned env["ROCR_VISIBLE_DEVICES"] = pinned