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