| """Frozen V4-family deployment variants: one checkpoint, fixture and CUDA reference each. |
| |
| All variants share the V4 contract (8K tokenizer, d384/L7/6 heads, SwiGLU 960, |
| context 128, tied head). Paths are relative to the repository root. |
| """ |
| from dataclasses import dataclass |
|
|
|
|
| @dataclass(frozen=True) |
| class Variant: |
| name: str |
| label: str |
| step: int |
| epoch: int |
| checkpoint: str |
| reference: str |
| cuda_scores: str |
| cuda_label: str |
| package_prefix: str |
| resource_tag: str |
|
|
| def package(self, kind): |
| """Deployment directory name, e.g. `int8-b32` -> tiny-ja-v4.3-epoch30-int8-b32-v1.""" |
| return f"{self.package_prefix}-{kind}-v1" |
|
|
|
|
| def _v43(epoch, step): |
| return Variant( |
| name=f"v43-epoch{epoch}", label=f"V4.3 epoch {epoch}", step=step, epoch=epoch, |
| checkpoint=f"artifacts/models/tiny-ja-v43-e8k-d384-l7/epoch{epoch}-deployment.pt", |
| reference=f"outputs/deployment/tiny-ja-v43-epoch{epoch}-fp32-input", |
| cuda_scores=f"outputs/evaluation-v43-20261020/epoch{epoch}", |
| cuda_label=f"V4.3 epoch{epoch} FP32 CUDA (2026-10-20)", |
| package_prefix=f"tiny-ja-v4.3-epoch{epoch}", resource_tag=f"V43E{epoch}") |
|
|
|
|
| VARIANTS = {variant.name: variant for variant in ( |
| Variant(name="v42-epoch20", label="V4.2 epoch 20", step=1081942, epoch=20, |
| checkpoint="artifacts/models/tiny-ja-v42-e8k-d384-l7/deployment.pt", |
| reference="outputs/deployment/tiny-ja-v42-epoch20-fp32-input", |
| cuda_scores="outputs/evaluation-v42-20261020", |
| cuda_label="V4.2 FP32 CUDA (2026-10-20)", |
| package_prefix="tiny-ja-v4.2-epoch20", resource_tag="V42"), |
| _v43(30, 3225888), |
| _v43(27, 2903298), |
| )} |
| DEFAULT = "v42-epoch20" |
|
|
