# okto_version: "1.2"
PROJECT "test_t5_custom_fields"
DESCRIPTION "Exemplo usando campos customizados no dataset (input_field e output_field)"

ENV {
    accelerator: "gpu"
    min_memory: "4GB"
    precision: "fp16"
    backend: "oktoseek"
    install_missing: true
}

DATASET {
    train: "dataset/train.jsonl"
    validation: "dataset/val.jsonl"
    format: "jsonl"
    type: "chat"
    # Campos customizados: define quais colunas usar do dataset
    input_field: "input"
    output_field: "target"
}

MODEL {
    base: "t5-small"
    device: "auto"
}

TRAIN {
    epochs: 3
    batch_size: 8
    learning_rate: 0.0001
    device: "auto"
}

EXPORT {
    format: ["okm", "safetensors"]
    path: "export/"
}