1 write to lm_head
Microsoft.ML.GenAI.Phi (1)
Phi3\Phi3ForCasualLM.cs (1)
35this.lm_head = new GenAILinear(config.HiddenSize, config.VocabSize, dtype: config.DType, hasBias: false);
1 reference to lm_head
Microsoft.ML.GenAI.Phi (1)
Phi3\Phi3ForCasualLM.cs (1)
45var logits = this.lm_head.forward(outputs.LastHiddenState);