configs/model_config.json

{
  "base_model": "Salesforce/codet5-small",
  "max_input_length": 512,
  "max_target_length": 256,
  "train_epochs": 3,
  "train_batch_size": 8,
  "learning_rate": 3e-05
}