Skip to content

rollout: add forge_load to replay dumped rollouts with sglang alive - #1959

Merged
zhuzilin merged 3 commits into
THUDM:mainfrom
ryang-max:feature/forge_load
May 27, 2026
Merged

zhuzilin merged 3 commits into
THUDM:mainfrom
ryang-max:feature/forge_load

Conversation

@ryang-max

@ryang-max ryang-max commented May 27, 2026

Copy link
Copy Markdown
Collaborator

Replays a dumped rollout .pt file via --rollout-function-path so memory-test runs can keep sglang/router/colocate offload-onload live while bypassing real generation. Unlike --load-debug-rollout-data this does NOT flip skip_sglang/debug_train_only.

Usage:

  --rollout-function-path slime.rollout.forge_load.generate_rollout
  --load-forge-rollout-data /path/to/rollout_data/0.pt            (literal)
  --load-forge-rollout-data /path/to/dumps/{rollout_id}.pt        (template)

ryang-max and others added 3 commits May 27, 2026 06:26
Replays a dumped rollout .pt file via --rollout-function-path so memory-test
runs can keep sglang/router/colocate offload-onload live while bypassing
real generation. Unlike --load-debug-rollout-data this does NOT flip
skip_sglang/debug_train_only.

Usage:
  --rollout-function-path slime.rollout.forge_load.generate_rollout
  --load-forge-rollout-data /path/to/rollout_data/0.pt            (literal)
  --load-forge-rollout-data /path/to/dumps/{rollout_id}.pt        (template)
@zhuzilin
zhuzilin merged commit 7ee0c14 into THUDM:main May 27, 2026
26 of 27 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants