fix: error when getting metric values no matching metrics found for s…

…0-rabbitmq-gooey-gpu-2ffacebook-2fseamless-m4t-v2-large
GooeyAI · Aug 12, 2024 · 6829a49 · 6829a49
1 parent 68dbd5c
commit 6829a49
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/chart/model-values.yaml b/chart/model-values.yaml
@@ -203,8 +203,8 @@ deployments:
 
   - name: "common-seamless"
     image: "crgooeyprodwestus1.azurecr.io/gooey-gpu-common:6"
+    limits_gpu: "10Gi"
     limits:
-      gpu: "10Gi"
       memory: "28Gi"  # (220 / 80) * 10
       cpu: "1"
     env:

diff --git a/chart/templates/deployment.yaml b/chart/templates/deployment.yaml
@@ -50,6 +50,10 @@ spec:
           - name: "RESOURCE_LIMITS_{{ $name | upper }}"
             value: "{{ $value }}"
           {{- end }}
+          {{- if .limits_gpu }}
+          - name: "RESOURCE_LIMITS_GPU"
+            value: "{{ .limits_gpu }}"
+          {{- end }}
         livenessProbe:
           exec:
             command: [ "bash", "-c", "celery inspect ping -d celery@$HOSTNAME" ]