1 interface inheriting from ICausalLMPipeline
Microsoft.ML.GenAI.Core (1)
Pipeline\CausalLMPipeline.cs (1)
17
public interface ICausalLMPipeline<out TTokenizer, out TModel> :
ICausalLMPipeline
1 implementation of ICausalLMPipeline
Microsoft.ML.GenAI.Core (1)
Pipeline\CausalLMPipeline.cs (1)
85
public class CausalLMPipeline :
ICausalLMPipeline
4 references to ICausalLMPipeline
Microsoft.ML.GenAI.Core (3)
Trainer\CasualLMSupervisedFineTuningTrainer.cs (3)
19
private readonly
ICausalLMPipeline
_pipeline;
21
public CasualLMSupervisedFineTuningTrainer(
ICausalLMPipeline
pipeline, ILogger<CasualLMSupervisedFineTuningTrainer>? logger = null)
28
public async IAsyncEnumerable<
ICausalLMPipeline
> TrainAsync(
Microsoft.ML.GenAI.Samples (1)
Llama\SFT_Llama_3_2_1B.cs (1)
60
await foreach (
var
p in sftTrainer.TrainAsync(input, option, default))