examples/variants_example.py

from open_mythos import (
    mythos_1b,
    OpenMythos,
)

cfg = mythos_1b()
model = OpenMythos(cfg)

total = sum(p.numel() for p in model.parameters())
print(f"Parameters: {total:,}")