1 write to _paddingIdx
Microsoft.ML.GenAI.LLaMA (1)
Module\LlamaModel.cs (1)
30
this.
_paddingIdx
= config.PadTokenId;
1 reference to _paddingIdx
Microsoft.ML.GenAI.LLaMA (1)
Module\LlamaModel.cs (1)
33
this.embed_tokens = nn.Embedding(config.VocabSize, config.HiddenSize, padding_idx: this.
_paddingIdx
, dtype: config.DType);