1 write to _vocabSize
Microsoft.ML.GenAI.LLaMA (1)
LlamaForCausalLM.cs (1)
31
_vocabSize
= config.VocabSize;
1 reference to _vocabSize
Microsoft.ML.GenAI.LLaMA (1)
LlamaForCausalLM.cs (1)
77
shiftLogits = shiftLogits.view(-1,
_vocabSize
);