1 write to NumKeyValueHeads
Microsoft.ML.GenAI.Mistral (1)
MistralConfig.cs (1)
26this.NumKeyValueHeads = 8;
2 references to NumKeyValueHeads
Microsoft.ML.GenAI.Mistral (2)
MistralDecoderLayer.cs (2)
97numKeyValueHeads: config.NumKeyValueHeads, 98numKeyValueGroups: config.NumAttentionHeads / config.NumKeyValueHeads,