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