5 references to CausalLMModelInput
Microsoft.ML.GenAI.Core (5)
Pipeline\CausalLMPipeline.cs (3)
146var input = new CausalLMModelInput(inputIds, attentionMask, pastKeyValuesLength: 0) 155var input = new CausalLMModelInput(inputIds[.., prevPos..curPos], attentionMask[.., prevPos..curPos], pastKeyValuesLength: prevPos) 311var input = new CausalLMModelInput(inputTensor, attentionMask, pastKeyValuesLength: 0)
Trainer\CasualLMSupervisedFineTuningTrainer.cs (1)
58var output = _pipeline.Model.forward(new CausalLMModelInput(inputIds, attentionMask: attentionMask, labels: labels, useCache: false));
Trainer\CausalLMDataset.cs (1)
93return new CausalLMModelInput(