| """Replaceable identity manifest for the current BarunAction-35M candidate.""" |
|
|
| from __future__ import annotations |
|
|
| from collections.abc import Mapping |
| from types import MappingProxyType |
|
|
| MODEL_NAME = "BarunAction-35M" |
| BASE_MODEL_NAME = "BarunLM-35M" |
| CANDIDATE_ID = "candidate-v2" |
| CANDIDATE_RUN_ID = "20260803-1845-mob-batch63-s17" |
| CANDIDATE_SELECTION_RUN_ID = "20260803-1845-mobile-followup-retry-s17" |
| CANDIDATE_ARM_ID = "batch63" |
| CANDIDATE_STEP = 126 |
| CANDIDATE_REPO_ID = "harrrshall/BarunAction-35M" |
| CANDIDATE_REVISION = "candidate-v2" |
| CANDIDATE_CHECKPOINT_SHA256 = MappingProxyType( |
| { |
| "barun_config.json": "9b3a1d71baa95a198744d250f9629231738d942570b8685c44307fd83dd33565", |
| "model.safetensors": "fdb95ccf58a095e0d321be998924318b35ee59a334f6dd97d8726d2cf80021d3", |
| "tokenizer.json": "70ded9605fccd09c2340ca7e225361eab0ae8b4dbbb0d6e26343ab5183979db6", |
| } |
| ) |
| CANDIDATE_MANIFEST_SHA256 = "c743ab7c4d33ae75c6b0aa4547458a961b92766da8fcf85fd148fda2ebb5530a" |
|
|
|
|
| def candidate_identity(checkpoint_sha256: Mapping[str, str]) -> tuple[str | None, str | None]: |
| """Label only the exact hash-pinned candidate; custom checkpoints remain unlabeled.""" |
|
|
| if dict(checkpoint_sha256) == dict(CANDIDATE_CHECKPOINT_SHA256): |
| return CANDIDATE_ID, CANDIDATE_RUN_ID |
| return None, None |
|
|
|
|
| __all__ = [ |
| "BASE_MODEL_NAME", |
| "CANDIDATE_ARM_ID", |
| "CANDIDATE_CHECKPOINT_SHA256", |
| "CANDIDATE_ID", |
| "CANDIDATE_MANIFEST_SHA256", |
| "CANDIDATE_REPO_ID", |
| "CANDIDATE_REVISION", |
| "CANDIDATE_RUN_ID", |
| "CANDIDATE_SELECTION_RUN_ID", |
| "CANDIDATE_STEP", |
| "MODEL_NAME", |
| "candidate_identity", |
| ] |
|
|