From 02d9f6178674d180f0178757f8792a5e6017515a Mon Sep 17 00:00:00 2001
From: Tyler Perkins <tyler@clortox.com>
Date: Sat, 30 Mar 2024 23:09:15 -0400
Subject: [PATCH] Update to nvidia runtime

---
 text-ui/text-ui-deployment.yaml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/text-ui/text-ui-deployment.yaml b/text-ui/text-ui-deployment.yaml
index f6a5c53..8a5eddc 100644
--- a/text-ui/text-ui-deployment.yaml
+++ b/text-ui/text-ui-deployment.yaml
@@ -13,6 +13,7 @@ spec:
       labels:
         app: text-generation-webui
     spec:
+      runtimeClassName: nvidia
       containers:
       - name: text-generation-webui
         image: atinoda/text-generation-webui:default-nvidia