kiln_ai.adapters.fine_tune.finetune_registry

 1from typing import Type
 2
 3from kiln_ai.adapters.fine_tune.base_finetune import BaseFinetuneAdapter
 4from kiln_ai.adapters.fine_tune.fireworks_finetune import FireworksFinetune
 5from kiln_ai.adapters.fine_tune.together_finetune import TogetherFinetune
 6from kiln_ai.adapters.fine_tune.vertex_finetune import VertexFinetune
 7from kiln_ai.adapters.ml_model_list import ModelProviderName
 8
 9finetune_registry: dict[ModelProviderName, Type[BaseFinetuneAdapter]] = {
10    ModelProviderName.fireworks_ai: FireworksFinetune,
11    ModelProviderName.together_ai: TogetherFinetune,
12    ModelProviderName.vertex: VertexFinetune,
13}