github-actions[bot]
54e6124925
version: sunnypilot v2026.001.000 (dev) date: 2026-05-07T23:07:19 master commit: c28eb958740187620f2282023b8f1997cf90f583
29 lines
1.3 KiB
Python
29 lines
1.3 KiB
Python
from openpilot.sunnypilot.models.helpers import get_active_bundle
|
|
from openpilot.sunnypilot.models.runners.model_runner import ModelRunner
|
|
from openpilot.sunnypilot.models.runners.tinygrad.tinygrad_runner import TinygradRunner, TinygradSplitRunner
|
|
from openpilot.sunnypilot.models.runners.constants import ModelType
|
|
|
|
|
|
def get_model_runner() -> ModelRunner:
|
|
"""
|
|
Factory function to create and return the appropriate ModelRunner instance.
|
|
|
|
Selects TinygradRunner, choosing TinygradSplitRunner if separate vision/policy
|
|
models are detected in the active bundle.
|
|
|
|
:return: An instance of a ModelRunner subclass (ONNXRunner, TinygradRunner, or TinygradSplitRunner).
|
|
"""
|
|
bundle = get_active_bundle()
|
|
if bundle and bundle.models:
|
|
model_types = {m.type.raw for m in bundle.models}
|
|
# Check if the bundle uses separate vision and policy models (legacy or new split format)
|
|
split_types = {ModelType.vision, ModelType.policy, ModelType.offPolicy, ModelType.onPolicy}
|
|
if model_types & split_types:
|
|
return TinygradSplitRunner()
|
|
# Otherwise, assume a single model (likely supercombo)
|
|
if bundle.models:
|
|
return TinygradRunner(bundle.models[0].type.raw)
|
|
|
|
# Default fallback to TinygradRunner with the supercombo type if bundle info is missing/incomplete
|
|
return TinygradRunner(ModelType.supercombo)
|