LLM

Onboarding an LLM task

Suppose you would like to onboard an LLM model for the following dataset:

Following is an example of how you could construct a fdl.ModelInfo object and onboard such a model.

PROJECT_ID = 'example_project'
DATASET_ID = 'dialogue_data'
MODEL_ID = 'llm_model'

dataset_info = client.get_dataset_info(
    project_id=PROJECT_ID,
    dataset_id=DATASET_ID
)

model_task = fdl.ModelTask.LLM

model_info = fdl.ModelInfo.from_dataset_info(
    dataset_info=dataset_info,
    dataset_id=DATASET_ID,
    model_task=model_task
)

client.add_model(
    project_id=PROJECT_ID,
    dataset_id=DATASET_ID,
    model_id=MODEL_ID,
    model_info=model_info
)