| """Phi-4-mini base-model preset (Microsoft). Auto-registers. | |
| Small-footprint default for laptop / single-GPU local fine-tuning. | |
| """ | |
| from __future__ import annotations | |
| from pydantic import BaseModel, ConfigDict | |
| class Phi4MiniPreset(BaseModel): | |
| model_config = ConfigDict(extra="forbid") | |
| hf_id: str = "microsoft/Phi-4-mini-instruct" | |
| chat_template: str = "hermes" | |
| max_seq_len: int = 16384 | |
| fsdp_layer_class: str = "Phi3DecoderLayer" | |
| moe: bool = False | |
| _PRESET = Phi4MiniPreset() | |
| def preset() -> Phi4MiniPreset: | |
| return _PRESET | |
| def _register() -> None: | |
| from mindxtrain.models.registry import register_preset | |
| register_preset(_PRESET.hf_id, _PRESET) | |
| register_preset("phi4_mini", _PRESET) | |
| _register() | |
| __all__ = ["Phi4MiniPreset", "preset"] | |