parameters method

  1. @override
List<Value> parameters()
override

Implementation

@override
List<Value> parameters() {
  return [
    ...selfAttention.parameters(),
    ...crossAttention.parameters(), // NEW
    ...ffn.parameters(),
    ...ln1.parameters(),
    ...ln2.parameters(), // NEW
    ...ln3.parameters(), // Renamed
  ];
}