Update to nvidia runtime

2024-03-30 23:09:15 -04:00 · 2024-03-30 23:09:15 -04:00 · 02d9f61786
commit 02d9f61786
parent 70685fb135
1 changed files with 1 additions and 0 deletions
--- 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