File size: 1,736 Bytes
99297d7 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 | """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"
|