OpenMythos / examples /variants_example.py
Kye Gomez
new examples folder
227dbb1
Raw History Blame Contribute Delete
186 Bytes
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:,}")