From d68f562606880ce88cb388a9aeaac78da515a144 Mon Sep 17 00:00:00 2001
From: Andre Manoel <165937436+andreatgretel@users.noreply.github.com>
Date: Wed, 7 Aug 2024 14:27:17 -0300
Subject: [PATCH] RDS-971: Test harness - fix GPT-X, add NavFT, ensure monitors
 are working

Co-authored-by: Matt Kornfield <kornfield@gretel.ai>
GitOrigin-RevId: a04ad3c0cae66787f779843c24d2e329b36c27a2
---
 src/gretel_trainer/benchmark/gretel/models.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/gretel_trainer/benchmark/gretel/models.py b/src/gretel_trainer/benchmark/gretel/models.py
index 6a92c142..d627c602 100644
--- 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