parameters method

@override List<Value> parameters()

Implementation

@override
List<Value> parameters() {
  return [
    ...tokenEmbeddings.expand((v) => v.values),
    ...positionEmbeddings.expand((v) => v.values),
    ...blocks.expand((b) => b.parameters()),
    ...finalLayerNorm.parameters(),
    ...lmHead.parameters(),
  ];
}