3 references to CausalLMModelInput
Microsoft.ML.GenAI.Core (3)
Pipeline\CausalLMPipeline.cs (3)
142
var input = new
CausalLMModelInput
(inputIds, attentionMask, pastKeyValuesLength: 0)
151
var input = new
CausalLMModelInput
(inputIds[.., prevPos..curPos], attentionMask[.., prevPos..curPos], pastKeyValuesLength: prevPos)
307
var input = new
CausalLMModelInput
(inputTensor, attentionMask, pastKeyValuesLength: 0)