[Feat][Helm] Add vllmConfig.revision for HuggingFace model version pinning#961
Conversation
There was a problem hiding this comment.
Code Review
This pull request introduces support for pinning Hugging Face model versions using a new revision configuration parameter in the Helm chart. It updates the documentation, values schema, templates, and adds corresponding unit tests. However, the unit tests in helm/tests/ray-cluster_test.yaml contain incorrect document indices (using index 1 instead of 2) when asserting the contents of the vllm-entrypoint.sh script inside the rendered ConfigMap, which will cause the tests to target the wrong document.
|
Could you fix the pre-commit issue when you have time? |
3f8cf23 to
d4fdb98
Compare
Fixed, although I don't see them executing again. |
d4fdb98 to
942f2fb
Compare
|
@ruizhang0101 it looks like the functionality test timeout for some reason |
ruizhang0101
left a comment
There was a problem hiding this comment.
Hi, could you put this in the model spec instead of vllm config?
modelSpec:
- name: "llama3"
repository: "vllm/vllm-openai"
tag: "latest"
modelURL: "meta-llama/Llama-3.1-8B-Instruct"
modelRevision: <COMMIT_HASH>942f2fb to
c766fd8
Compare
…nning Add a `revision` field to `vllmConfig` that maps to vLLM's `--revision` flag, allowing operators to pin deployed models to a specific HuggingFace Hub branch, tag, or commit hash. Without this, vLLM always pulls the latest commit on the default branch, which can silently change model behaviour across re-deploys. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com> Signed-off-by: Roberto Barreda <roberto.barreda@midokura.com>
Signed-off-by: Roberto Barreda <roberto.barreda@midokura.com>
4c15218 to
a827843
Compare
Move revision pinning from vllmConfig.revision to modelRevision at the model spec level, making it a first-class field on each model entry. - Add modelRevision to modelSpec defaults and schema - Update deployment-vllm-multi and ray-cluster templates to read $modelSpec.modelRevision - Update tests to use modelRevision instead of vllmConfig.revision - Update docs to reference modelRevision - Add llama3 example entry to values-01 with pinned revision modelRevision for meta-llama/Llama-3.1-8B-Instruct obtained via: curl -s https://huggingface.co/api/models/meta-llama/Llama-3.1-8B-Instruct | yq -p json '.sha' Signed-off-by: Roberto Barreda <roberto.barreda@midokura.com>
a827843 to
e7665f3
Compare
Add a
modelRevisionfield to the model spec that maps to vLLM's--revisionflag, allowing operators to pin deployed models to a specific HuggingFace Hub branch, tag, or commit hash.Without this, vLLM always pulls the latest commit on the default branch, which can silently change model behaviour across re-deploys.
modelRevisionis a first-class field on each model spec entry (not nested undervllmConfig), making it discoverable and consistent with other model-level fields.Changes:
helm/values.yaml+values.schema.json: addmodelRevisionfield to model spec (empty string default = no pinning), remove oldvllmConfig.revisionhelm/templates/deployment-vllm-multi.yaml: emit--revision <value>when$modelSpec.modelRevisionis sethelm/templates/ray-cluster.yaml: same for Ray cluster deploymentshelm/tests/: positive and negative unit tests for both deployment pathsdocs/source/deployment/helm.rst: documentmodelRevisioninstead ofrevision.github/values-01-2pods-minimal-example.yaml: add llama3 example with pinnedmodelRevisionmodelRevision for
meta-llama/Llama-3.1-8B-Instructobtained via:curl -s https://huggingface.co/api/models/meta-llama/Llama-3.1-8B-Instruct | yq -p json '.sha'