diff --git a/charts/cog-ai-model/templates/deployment.yaml b/charts/cog-ai-model/templates/deployment.yaml
index 424188a..600a488 100644
--- a/charts/cog-ai-model/templates/deployment.yaml
+++ b/charts/cog-ai-model/templates/deployment.yaml
@@ -23,7 +23,9 @@ spec:
         {{- toYaml . | nindent 8 }}
         {{- end }}
     spec:
-      runtimeClassName: nvidia
+      {{- if not .Values.runtimeClassName }}
+      runtimeClassName: {{ .Values.runtimeClassName }}
+      {{- end }}
       {{- with .Values.imagePullSecrets }}
       imagePullSecrets:
         {{- toYaml . | nindent 8 }}
diff --git a/charts/cog-ai-model/values.yaml b/charts/cog-ai-model/values.yaml
index 62cbf20..be8c318 100644
--- a/charts/cog-ai-model/values.yaml
+++ b/charts/cog-ai-model/values.yaml
@@ -16,6 +16,8 @@ config:
   modelLocalDir: "/var/huggingface/cache"
   modelMountDir: "/root/.cache/huggingface"
 
+runtimeClassName: nvidia
+
 imagePullSecrets: []
 nameOverride: ""
 fullnameOverride: ""
@@ -153,3 +155,4 @@ readinessProbe:
 
 envs: {}
 envsFrom: {}
+