github-actions[bot]
f9fcc7adab
date: 2026-06-28T09:48:35 master commit: da6313dbe95b3f24bb5d8018b0e5f950f5823ca7
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)
|