3 instantiations of CausalLMModelInput
Microsoft.ML.GenAI.Core (3)
17 references to CausalLMModelInput
Microsoft.ML.GenAI.Core (7)
Pipeline\CausalLMPipeline.cs (7)
19where TModel : nn.Module<CausalLMModelInput, CausalLMModelOutput>
66where TModel : nn.Module<CausalLMModelInput, CausalLMModelOutput>
94nn.Module<CausalLMModelInput, CausalLMModelOutput> model,
114public nn.Module<CausalLMModelInput, CausalLMModelOutput> Model { get; }
142var input = new CausalLMModelInput(inputIds, attentionMask, pastKeyValuesLength: 0)
151var input = new CausalLMModelInput(inputIds[.., prevPos..curPos], attentionMask[.., prevPos..curPos], pastKeyValuesLength: prevPos)
297var input = new CausalLMModelInput(inputTensor, attentionMask, pastKeyValuesLength: 0);
Microsoft.ML.GenAI.LLaMA (4)
Microsoft.ML.GenAI.Phi (6)