1 write to embed_tokens
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);
2 references to embed_tokens
Microsoft.ML.GenAI.LLaMA (2)
Module\LlamaModel.cs (2)
50
public Embedding Embedding => this.
embed_tokens
;
82
inputsEmbeds = this.
embed_tokens
.forward(inputIds);