Repository navigation
feat(slurm): support hosted models through native DD provider settings - #1080
Draft
dhruvnathawani wants to merge 7 commits into
Draft
dhruvnathawani wants to merge 7 commits into
dhruvnathawani wants to merge 7 commits into
Conversation
Allow hosted endpoints alongside managed models and run hosted-only recipes in CPU allocations. Resolve credentials at runtime and preserve native Data Designer model settings and asynchronous execution. Reuse shard, retry, resume, and collection contracts while preserving existing local-only plan serialization. Add configuration examples, documentation, and native client and allocation lifecycle coverage. Closes #1079 Signed-off-by: Dhruv Nathawani <[email protected]>
Signed-off-by: Dhruv Nathawani <[email protected]>
Contributor
|
All contributors have signed the DCO ✍️ ✅ |
Contributor
|
Fern preview: https://nvidia-preview-pr-1080.docs.buildwithfern.com/nemo/datadesigner
|
dhruvnathawani
force-pushed
the
dhruv/slurm-hosted-providers
branch
from
October 23, 2026 05:52
2f19916 to
dfcbbf0
Compare
Signed-off-by: Dhruv Nathawani <[email protected]>
Signed-off-by: Dhruv Nathawani <[email protected]>
2 tasks done
Signed-off-by: Dhruv Nathawani <[email protected]>
Signed-off-by: Dhruv Nathawani <[email protected]>
…tings Signed-off-by: Dhruv Nathawani <[email protected]>
This branch has not been deployed
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
📋 Summary
Slurm recipes can now use hosted models configured through DD's existing
model_providers.yaml, alongside managed vLLM deployments or in hosted-only CPU allocations. The Slurm run file describes only the models deployed locally; there is no parallel provider schema orinvocation.model_providerssetting.Targets
feat/slurm-execution. Live qualification combines head774774487da6with the independent vLLM version-gate fix in #1086 (3442c185abac). This PR remains draft for review.🔗 Related Issue
Closes #1079. Implements the native-provider ownership and snapshot flow proposed in Nabin's comment.
🔄 Changes
--gres=none, without GPU discovery, serving images, containers, or proxies. Add per-nodesubmission.memoryfor generation/collection.Async execution does not guarantee continuous GPU utilization. Local GPUs remain allocated during hosted requests; separate stages remain appropriate when hosted processing is substantial.
🧪 Testing
make check-all, changed-file Ruff checks, andgit diff --checkpassed.pass, then failed during Parquet consolidation. CPU replay with the export fix preserved all four saved records exactly; the original failed attempt remains preserved.pass, and all four records published and collected. Generation completed with exit code 0; collection also completed with exit code 0 on a CPU allocation with zero GPUs.The small fixture tests provider routing and publication. It does not qualify Gym SQL execution, production SQL data quality, sustained GPU utilization, or a full dataset/SFT rerun.
✅ Checklist