task:
  name: "styling"
  type: "style_transfer"

data:
  source: "custom"
  input_field: "text"
  style_field: "style"
  max_length: 256
  train_split: 0.8
  validation_split: 0.1
  test_split: 0.1

model:
  name: "t5-base"
  max_length: 256

training:
  num_epochs: 3
  batch_size: 16
  learning_rate: 3e-5
  weight_decay: 0.01
  warmup_ratio: 0.1
  lr_scheduler_type: "linear"

inference:
  batch_size: 32
  max_new_tokens: 128
  temperature: 0.8