feat(charts): Add vllm-omni container support in API deployment

Signed-off-by: Victor Hang <vhvictorhang@gmail.com>

Author: Banh-Canh

charts/azimuth-llm/templates/api/deployment.yml

@@ -26,13 +26,21 @@ spec:
         {{- else }}
         image: "{{ .Values.api.image.containerImage }}:{{ .Values.api.image.version }}"
         {{- end }}
+        {{- if eq .Values.api.image.containerImage "vllm/vllm-omni" }}
+        command:
+        - vllm
+        - serve
+        {{- end }}
         ports:
         - name: api
           containerPort: 8000
         volumeMounts:
         - name: data
           mountPath: /root/.cache/huggingface
         args:
+          {{- if eq .Values.api.image.containerImage "vllm/vllm-omni" }}
+          - --omni
+          {{- end }}
           {{- if semverCompare "<v0.13.0" .Values.api.image.version }}
           - --model
           {{- end }}