|
|
# okto_version: "1.2"
|
|
|
PROJECT "pizzaria_chatbot"
|
|
|
DESCRIPTION "Chatbot de pizzaria com campos de contexto (menu, drinks, etc.)"
|
|
|
|
|
|
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 principais
|
|
|
input_field: "input"
|
|
|
output_field: "target"
|
|
|
|
|
|
# Campos de contexto que serão incluídos automaticamente no prompt
|
|
|
context_fields: ["menu", "drinks", "promotions"]
|
|
|
}
|
|
|
|
|
|
MODEL {
|
|
|
base: "t5-small"
|
|
|
device: "auto"
|
|
|
}
|
|
|
|
|
|
TRAIN {
|
|
|
epochs: 3
|
|
|
batch_size: 8
|
|
|
learning_rate: 0.0001
|
|
|
device: "auto"
|
|
|
}
|
|
|
|
|
|
EXPORT {
|
|
|
format: ["okm", "safetensors"]
|
|
|
path: "export/"
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
