OpenMythos/variants_example.py

11 lines
186 B
Python

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:,}")