| """Register V5 only in its explicit entry points; V1/V2/V4 registrations unchanged.""" | |
| from vimeml.training.model_factory import ARCHITECTURES | |
| from vimeml.training.model_v5 import GPTV5Config, TinyGPTV5 | |
| ARCHITECTURE = "tiny_gpt_v5" | |
| FORMAT = "vimeml_tiny_gpt_v5" | |
| def register_v5(): | |
| definition = (GPTV5Config, TinyGPTV5, FORMAT) | |
| if ARCHITECTURE in ARCHITECTURES and ARCHITECTURES[ARCHITECTURE] != definition: | |
| raise ValueError("Conflicting V5 registration") | |
| ARCHITECTURES[ARCHITECTURE] = definition | |
| register_v5() | |