Luna Studio SDK

run_steps:
  - data_generation
  - training

pipeline_provider: "local"
metric_name: "custom"

pipeline:
  pipeline_name: "splunk-ao-luna-ft-sample-run"

data_generation:
  metric:
    name: off_topic_detection
    description:
    type: binary
    input_format: single
    llmaj_source_prompt: "Detect if the user query is off-topic."
  source_data:
    dataset:
      source_type: "huggingface"
      huggingface:
        name: "rungalileo/off-topic-sample"
  generation:
    total_examples: 50
  output:
    push_to_hub: false
    local_path: "./data/generated_data"
    dataset:
      repo_name: off-topic-sample-training
  data_quality_metrics:
    enabled: false

training:
  metric:
    type: boolean
  dataset:
    name: off-topic-sample-training
    local: true
    local_path: "./data/generated_data"
  prompt_template:
  output:
    model_name: "off-topic-detection-splunk-ao-luna-ft"
    push_to_hub: false
    local_path: finetuned_models
  training:
    num_train_epochs: 1
    use_wandb: false