__init__.py — PI0 模型子模块的包初始化文件¶
文件路径:
verl/experimental/vla/models/pi0_torch/__init__.py
文件概述¶
PI0 模型子模块的包初始化文件,导出核心类。
from .configuration_pi0_torch import PI0TorchConfig
from .modeling_pi0_torch import PI0ForActionPrediction
导出组件¶
| 组件 | 说明 |
|---|---|
PI0TorchConfig |
PI0 模型配置 |
PI0ForActionPrediction |
PI0 动作预测模型 |
小结¶
导出 PI0 模型的配置类和模型类,供 register_vla_models.py 注册使用。