vLLMWorker

class torchrl.modules.llm.vLLMWorker(*args, **kwargs)

Ray 的 vLLM 工作器。

vLLMParameterServer 将始终采用此工作器初始化的无状态进程组中的 rank 0。LLM 类相关的 tp ranks 将在 [1, tp_size] 范围内。

check_weights_changed()

检查权重是否更新为 0。