From 49bc1904035a4abc1de4611ca02439138d24cb0a Mon Sep 17 00:00:00 2001
From: Nina Xu <19981858+nina-xu@users.noreply.github.com>
Date: Mon, 3 Aug 2026 07:36:54 -0700
Subject: [PATCH] update dp configs to new shape

Signed-off-by: Nina Xu <19981858+nina-xu@users.noreply.github.com>
---
 script/slurm/configs/mistral-dp.yaml   | 3 ++-
 script/slurm/configs/smollm3-dp.yaml   | 3 ++-
 script/slurm/configs/tinyllama-dp.yaml | 3 ++-
 3 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/script/slurm/configs/mistral-dp.yaml b/script/slurm/configs/mistral-dp.yaml
index 27511e27e..e5cc13db0 100644
--- a/script/slurm/configs/mistral-dp.yaml
+++ b/script/slurm/configs/mistral-dp.yaml
@@ -8,5 +8,6 @@ training:
 privacy:
   dp_enabled: true
 generation:
-  use_structured_generation: true
+  structured_generation:
+    enabled: true
   num_records: 3000
diff --git a/script/slurm/configs/smollm3-dp.yaml b/script/slurm/configs/smollm3-dp.yaml
index a72454739..f532c0373 100644
--- a/script/slurm/configs/smollm3-dp.yaml
+++ b/script/slurm/configs/smollm3-dp.yaml
@@ -8,5 +8,6 @@ training:
 privacy:
   dp_enabled: true
 generation:
-  use_structured_generation: true
+  structured_generation:
+    enabled: true
   num_records: 3000
diff --git a/script/slurm/configs/tinyllama-dp.yaml b/script/slurm/configs/tinyllama-dp.yaml
index 08123fd86..8560e4dbe 100644
--- a/script/slurm/configs/tinyllama-dp.yaml
+++ b/script/slurm/configs/tinyllama-dp.yaml
@@ -8,5 +8,6 @@ training:
 privacy:
   dp_enabled: true
 generation:
-  use_structured_generation: true
+  structured_generation:
+    enabled: true
   num_records: 3000