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.openai_finetune import OpenAIFinetune 5from kiln_ai.adapters.ml_model_list import ModelProviderName 6 7finetune_registry: dict[ModelProviderName, Type[BaseFinetuneAdapter]] = { 8 ModelProviderName.openai: OpenAIFinetune, 9}
finetune_registry: dict[kiln_ai.adapters.ml_model_list.ModelProviderName, typing.Type[kiln_ai.adapters.fine_tune.base_finetune.BaseFinetuneAdapter]] =
{<ModelProviderName.openai: 'openai'>: <class 'kiln_ai.adapters.fine_tune.openai_finetune.OpenAIFinetune'>}