github-actions[bot]
de77c89cf6
version: sunnypilot v2026.002.000 (staging) date: 2026-07-21T07:44:02 master commit: 2c334ede443d7391d27575af8c854a095ba702a8
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)
|