RDS-971: Test harness - fix GPT-X, add NavFT, ensure monitors are wor…

…king Co-authored-by: Matt Kornfield <[email protected]> GitOrigin-RevId: a04ad3c0cae66787f779843c24d2e329b36c27a2
gretelai · Aug 7, 2024 · d68f562 · d68f562
1 parent 7b8e98f
commit d68f562
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/gretel_trainer/benchmark/gretel/models.py b/src/gretel_trainer/benchmark/gretel/models.py
@@ -86,6 +86,10 @@ class GretelTabularDP(GretelModel):
     config = "synthetics/tabular-differential-privacy"
 
 
+class GretelNavigatorFT(GretelModel):
+    config = "synthetics/navigator-ft"
+
+
 class _GretelModelWithOverrides(GretelModel):
     _delegate: GretelModel
     _name: str