跳转至

__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 注册使用。