1 write to model
Microsoft.ML.GenAI.LLaMA (1)
LlamaForCausalLM.cs (1)
33
model
= new LlamaModel(config, device);
3 references to model
Microsoft.ML.GenAI.LLaMA (3)
LlamaForCausalLM.cs (3)
53
var embeddingWeight =
model
.Embedding.state_dict();
56
this.lm_head.to(device:
model
.Embedding.weight!.device);
63
var outputs = this.
model
.forward(input);