[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
2026-04-24 15:50:40 +00:00 · 2026-04-24 15:50:40 +00:00 · 56c95c69fb
commit 56c95c69fb
parent eff55fb8db
1 changed files with 1 additions and 0 deletions
--- 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