2 references to Embedding
Microsoft.ML.GenAI.LLaMA (2)
LlamaForCausalLM.cs (2)
53
var embeddingWeight = model.
Embedding
.state_dict();
56
this.lm_head.to(device: model.
Embedding
.weight!.device);