From 331659f47ff03fe86a1dce078aa3ec1a76e509e0 Mon Sep 17 00:00:00 2001
From: Masaki Yatsu <yatsu@yatsu.info>
Date: Wed, 3 Dec 2025 15:55:37 +0900
Subject: [PATCH] fix(ollama): set GPU runtime class name

---
 ollama/values.gomplate.yaml | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/ollama/values.gomplate.yaml b/ollama/values.gomplate.yaml
index 99e0ff8..22462c4 100644
--- a/ollama/values.gomplate.yaml
+++ b/ollama/values.gomplate.yaml
@@ -1,3 +1,7 @@
+{{- if eq .Env.OLLAMA_GPU_ENABLED "true" }}
+runtimeClassName: "nvidia"
+{{- end }}
+
 ollama:
   gpu:
     enabled: {{ if eq .Env.OLLAMA_GPU_ENABLED "true" }}true{{ else }}false{{ end }}