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