File size: 535 Bytes
99297d7 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | """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()
|