__init__.py — Rollout 模块初始化

文件概述

导出 Rollout 模块的核心类和工厂函数。

导出内容

from .base import BaseRollout, get_rollout_class
from .hf_rollout import HFRollout
from .naive import NaiveRollout
from .replica import RolloutReplica

__all__ = ["BaseRollout", "NaiveRollout", "HFRollout", "get_rollout_class", "RolloutReplica"]

• BaseRollout: Rollout 基类，定义生成接口
• get_rollout_class: 工厂函数，根据配置返回对应的 Rollout 类
• NaiveRollout: 基于 HuggingFace 的简单 Rollout
• HFRollout: HuggingFace Rollout 实现
• RolloutReplica: 多副本 Rollout 管理

小结

提供 Rollout 模块的统一导入入口，包含多种推理后端的 Rollout 实现。