File size: 840 Bytes
5df2c77 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
# 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/"
}
|