From 9b3c4f8d50020ded33621d7d8d05a59c7f8eb110 Mon Sep 17 00:00:00 2001
From: Eda Z <36808054+edamamez@users.noreply.github.com>
Date: Tue, 3 Sep 2024 15:38:27 -0700
Subject: [PATCH] Update get_default_finetune_args.py

---
 .../text2sql_memory_tuning/util/get_default_finetune_args.py  | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/recipes/3p_integrations/lamini/text2sql_memory_tuning/util/get_default_finetune_args.py b/recipes/3p_integrations/lamini/text2sql_memory_tuning/util/get_default_finetune_args.py
index 6cddb41a6..3d24dae56 100644
--- a/recipes/3p_integrations/lamini/text2sql_memory_tuning/util/get_default_finetune_args.py
+++ b/recipes/3p_integrations/lamini/text2sql_memory_tuning/util/get_default_finetune_args.py
@@ -1,7 +1,7 @@
 def get_default_finetune_args():
     return {
-        "learning_rate": 3e-4,
-        "max_steps": 360,
+        "learning_rate": 0.0003,
+        "max_steps": 60,
         "early_stopping": False,
         "load_best_model_at_end": False,
         "peft_args": {"r_value": 32},