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