File size: 905 Bytes
dfb775d | 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 | """GLM-5.1 base-model preset (Z.ai, 754B/40B-A MoE, MIT-licensed).
Specialist track only — see LICENSE-MIT-upstream-glm51 for upstream license
terms. Auto-registers with `ModelRegistry` at import time.
"""
from __future__ import annotations
from pydantic import BaseModel, ConfigDict
class Glm51Preset(BaseModel):
model_config = ConfigDict(extra="forbid")
hf_id: str = "zai-org/GLM-5.1-Base"
chat_template: str = "qwen3_reasoning"
max_seq_len: int = 32768
fsdp_layer_class: str = "GLMDecoderLayer"
moe: bool = True
_PRESET = Glm51Preset()
def preset() -> Glm51Preset:
return _PRESET
# Auto-register so `ModelRegistry.presets()` knows about us.
def _register() -> None:
from mindxtrain.models.registry import register_preset
register_preset(_PRESET.hf_id, _PRESET)
register_preset("glm51", _PRESET)
_register()
__all__ = ["Glm51Preset", "preset"]
|