| """DeepSeek V3.2 base-model preset. | |
| Reasoning-tuned checkpoint with `<think>...</think>` framing. Auto-registers. | |
| """ | |
| from __future__ import annotations | |
| from pydantic import BaseModel, ConfigDict | |
| class DeepSeekV32Preset(BaseModel): | |
| model_config = ConfigDict(extra="forbid") | |
| hf_id: str = "deepseek-ai/DeepSeek-V3.2" | |
| chat_template: str = "deepseek_r1" | |
| max_seq_len: int = 65536 | |
| fsdp_layer_class: str = "DeepseekV3DecoderLayer" | |
| moe: bool = True | |
| _PRESET = DeepSeekV32Preset() | |
| def preset() -> DeepSeekV32Preset: | |
| return _PRESET | |
| def _register() -> None: | |
| from mindxtrain.models.registry import register_preset | |
| register_preset(_PRESET.hf_id, _PRESET) | |
| register_preset("deepseek_v32", _PRESET) | |
| _register() | |
| __all__ = ["DeepSeekV32Preset", "preset"] | |