Skip to content

Conversation

pradeepfn
Copy link
Contributor

@pradeepfn pradeepfn commented Sep 2, 2025

Policy engine introduce a Policy service. This broke the existing vllm + torchstore example script.
This diff fixes it.

  1. Only the single worker case is fixed here.
    2 Will fix the multi-worker case in followup PR.

@meta-cla meta-cla bot added the CLA Signed This label is managed by the Meta Open Source bot. label Sep 2, 2025
@pradeepfn pradeepfn changed the title [DRAFT] VLLM + torchstore integration test upgrade to match the new Policy APIs. VLLM + torchstore integration test upgrade to match the new Policy APIs. Sep 4, 2025
return self.weights_version

@endpoint
async def get_model_params(self) -> Dict[str, torch.Tensor]:
Copy link
Member

Choose a reason for hiding this comment

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

If it's only for testing purposes, can you underscore it please?

@pradeepfn pradeepfn merged commit 75b5fd6 into meta-pytorch:main Sep 4, 2025
4 of 5 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CLA Signed This label is managed by the Meta Open Source bot.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants