def Cfg.heads

bend-ml-llama@0.2.1.0 · main.bend:138

import Base

138# the query heads: kv groups of g
139def Cfg.heads(+c: Cfg) -> Nat:
140  Nat.mul(Cfg.kv(c), Cfg.g(c))