mawei/StarPilot
1
0
Fork 0
mirror of https://github.com/firestar5683/StarPilot.git synced 2026-08-21 00:03:45 +08:00
Code Issues Packages Projects Releases Wiki Activity

RL code

Browse Source
This commit is contained in:
firestar5683
2026-04-28 21:14:00 -05:00
parent d547e3632d
commit 422d92f0ea
11 changed files with 54 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files
starpilot/assets/model_manager.py
+2 -2
View File
@@ -18,7 +18,7 @@ from openpilot.starpilot.common.starpilot_utilities import delete_file
from openpilot.starpilot.common.starpilot_variables import MODELS_PATH
MANIFEST_CANDIDATES = ("v21",)
TINYGRAD_VERSIONS = {"v8", "v9", "v10", "v11", "v12"}
TINYGRAD_VERSIONS = {"v8", "v9", "v10", "v11", "v12", "v13"}
DEFAULT_MODEL_KEY = "sc2"
MODEL_KEY_CANONICAL_MAP = {
"sc": DEFAULT_MODEL_KEY,
@@ -192,7 +192,7 @@ class ModelManager:
f"{model_key}_driving_vision_metadata.pkl",
]
if model_version == "v12":
if model_version in {"v12", "v13"}:
filenames += [
f"{model_key}_driving_off_policy_tinygrad.pkl",
f"{model_key}_driving_off_policy_metadata.pkl",
starpilot/common/starpilot_variables.py
+1 -1
View File
@@ -972,7 +972,7 @@ class StarPilotVariables:
if isinstance(toggle.model_version, bytes):
toggle.model_version = toggle.model_version.decode("utf-8", "ignore")
toggle.classic_model = toggle.model_version in {"v1", "v2", "v3", "v4"}
toggle.tinygrad_model = toggle.model_version in {"v8", "v9", "v10", "v11", "v12"}
toggle.tinygrad_model = toggle.model_version in {"v8", "v9", "v10", "v11", "v12", "v13"}
toggle.tomb_raider = toggle.model == "space-lab"
toggle.model_ui = self.get_value("ModelUI")
starpilot/system/the_pond/the_pond.py
+2 -2
View File
@@ -4340,14 +4340,14 @@ def setup(app):
if f"{model_key}.thneed" in on_disk_files:
return True
if model_version in ("v8", "v9", "v10", "v11", "v12"):
if model_version in ("v8", "v9", "v10", "v11", "v12", "v13"):
required_files = {
f"{model_key}_driving_policy_tinygrad.pkl",
f"{model_key}_driving_vision_tinygrad.pkl",
f"{model_key}_driving_policy_metadata.pkl",
f"{model_key}_driving_vision_metadata.pkl",
}
if model_version == "v12":
if model_version in ("v12", "v13"):
required_files |= {
f"{model_key}_driving_off_policy_tinygrad.pkl",
f"{model_key}_driving_off_policy_metadata.pkl",