| """Register V4 only in its explicit entry points; frozen V1/V2 factory unchanged.""" | |
| from vimeml.training.model_factory import ARCHITECTURES | |
| from vimeml.training.model_v4 import GPTV4Config, TinyGPTV4 | |
| ARCHITECTURE = "tiny_gpt_v4" | |
| FORMAT = "vimeml_tiny_gpt_v4" | |
| def register_v4(): | |
| definition = (GPTV4Config, TinyGPTV4, FORMAT) | |
| if ARCHITECTURE in ARCHITECTURES and ARCHITECTURES[ARCHITECTURE] != definition: | |
| raise ValueError("Conflicting V4 registration") | |
| ARCHITECTURES[ARCHITECTURE] = definition | |
| register_v4() | |