Conversation
tjruwase
reviewed
Jun 30, 2026
tjruwase
reviewed
Jun 30, 2026
tjruwase
reviewed
Jun 30, 2026
Contributor
Author
|
Hi @tjruwase I have changed according to your first and second comments. For the third comment, I left an explaination. Thanks! |
Entry point, configs, data, and tests for on-policy distillation using DeepSpeed's hybrid engine rollout and vLLM backend. Signed-off-by: Guokai Ma <guokai.ma@intel.com> Signed-off-by: Guokai Ma <guokai.ma@gmail.com>
…m_dtype to engine_dtype Signed-off-by: Guokai Ma <guokai.ma@gmail.com>
| ``` | ||
| examples/opsd/ | ||
| ├── main.py # entry point (deepspeed launcher) | ||
| ├── opsd/ |
Contributor
There was a problem hiding this comment.
It seems this folder structure exists in DS not DSE, right?
Contributor
Author
There was a problem hiding this comment.
yes, the folder should start from training/opsd, I'll update them
…DeepSpeed - Delete vLLM configs, scripts (opsd_vllm_disjoint.json, smoke_vllm.json, train_opsd_vllm.sh) - Add trainer.py, config.py, losses.py, utils.py (moved from DeepSpeed) - Add benchmarks/ (5 hybrid engine benchmarks moved from DeepSpeed) - Update main.py imports (trainer, config now local) - Update test imports (losses, utils now local) - Rewrite README (remove all vLLM sections) Signed-off-by: Guokai Ma <guokai.ma@gmail.com>
Signed-off-by: Guokai Ma <guokai.ma@gmail.com>
…ants from JSON - Subclass DeepSpeed's RolloutConfig to add temperature/top_p/etc - Remove weight_sync_interval from JSON configs (vLLM remnant) Signed-off-by: Guokai Ma <guokai.ma@gmail.com>
…per fix - Remove 14B/multi-GPU benchmarks (bench_14b_rollout, bench_autotp_gc, bench_hybrid_tp, bench_hybrid_tp_opt) - Fix bench_decode_1p1r: wrap model for HybridEngineRollout - Add --graph-capture CLI flag Signed-off-by: Guokai Ma <guokai.ma@gmail.com>
…nfigs temperature=0 causes logits/0 = inf → NaN loss. The correct default for knowledge distillation is temperature=1.0 (standard softmax). Signed-off-by: Guokai Ma <guokai.ma@gmail.com>
Signed-off-by: Guokai Ma <guokai.ma@gmail.com>
| for p in model.parameters(): | ||
| p.requires_grad_(False) | ||
|
|
||
| if cfg.offload_to_cpu: |
Contributor
There was a problem hiding this comment.
Does this mean deepspeed is only enabled when offload_to_cpu==True?
Why not
- Always enable deepspeed
ds_config["zero_optimization"]["offload_param"]["device"] == "cpu" ifcfg.offload_to_cpuelse None
Signed-off-by: Guokai Ma <guokai.ma@gmail.com>
Signed-off-by: Guokai Ma <guokai.ma@gmail.com>
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
3 participants
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.
This PR moved example code from @PKUWZP 's OPSD PR in DeepSpeed (deepspeedai/DeepSpeed#8027).
Note the basic OPSD infrastructures (OPSD trainer, rollout engines) are still in the original PR and need to be merged seperately. This PR will work when the original PR merged.