Skip to content

[skyrl-train][Fix] sleep only if colocated#595

Merged
SumanthRH merged 2 commits intomainfrom
fix-sleep
Oct 29, 2025
Merged

[skyrl-train][Fix] sleep only if colocated#595
SumanthRH merged 2 commits intomainfrom
fix-sleep

Conversation

@SumanthRH
Copy link
Member

No description provided.

x
Signed-off-by: SumanthRH <sumanthrh99@gmail.com>
Signed-off-by: SumanthRH <sumanthrh99@gmail.com>
Copy link
Contributor

@gemini-code-assist gemini-code-assist bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

This pull request correctly fixes an issue where the inference engine was being put to sleep even when not co-located with other models. By adding a check for self.colocate_all, the change prevents unnecessary and inefficient offloading of the inference model from the GPU in non-colocated setups. The logic is sound and addresses the issue described in the title. I have one minor suggestion to improve code style.

@SumanthRH SumanthRH merged commit 55fce93 into main Oct 29, 2025
1 check passed
li-boxuan pushed a commit to li-boxuan/SkyRL that referenced this pull request Nov 23, 2025
Signed-off-by: SumanthRH <sumanthrh99@gmail.com>
@tyler-griggs tyler-griggs deleted the fix-sleep branch January 9, 2026 19:33
dzorlu pushed a commit to fleet-ai/SkyRL that referenced this pull request Feb 4, 2026
Signed-off-by: SumanthRH <sumanthrh99@gmail.com>
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