legaldoc_summarizer/config.yaml

project: "LegalDocSummarizer"
base_model: "google/flan-t5-base"
finetuned_model: ""        # Optional: HF repo ID for a private/public fine-tuned model
finetuned_local_dir: "models/legaldoc_summarizer"
dataset_name: "cuad"   # Contract Understanding Atticus Dataset
train:
  epochs: 3
  batch_size: 4
  lr: 2e-4
  lora_r: 8
  lora_alpha: 16
  lora_dropout: 0.05
evaluate:
  metrics: ["rouge", "bleu", "factuality"]
demo:
  max_new_tokens: 300