← Về thư mục
verl/workers/rollout/hf_rollout.py
- HFRollout · class · L35-L140 — class HFRollout(BaseRollout)
- init · method · L37-L40 — def init(self, module: nn.Module, config)
- generate_sequences · method · L42-L48 — def generate_sequences(self, prompts: DataProto) -> DataProto
- _generate_minibatch · method · L51-L140 — def _generate_minibatch(self, prompts: DataProto) -> DataProto