From bb314fa60de4e348407c70294cebec1d9d6f27b0 Mon Sep 17 00:00:00 2001
From: erew123 <35898566+erew123@users.noreply.github.com>
Date: Sun, 20 Oct 2024 07:26:30 +0100
Subject: [PATCH] minor additions

Close GPTTrainerConfig( on line 965
analytics_enabled=False for Gradio (stop it complaining about new versions of Gradio)
demo.queue().launch( line 2387 to launch Gradio.
---
 finetune.py | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/finetune.py b/finetune.py
index d5742fe..f675c26 100644
--- a/finetune.py
+++ b/finetune.py
@@ -920,6 +920,7 @@ def train_gpt(language, num_epochs, batch_size, grad_acumm, train_csv, eval_csv,
 
 
     print(f"[FINETUNE] Learning Scheduler {lr_scheduler}, params {lr_scheduler_params}")
+
     # training parameters config
     config = GPTTrainerConfig(
         epochs=num_epochs,
@@ -952,6 +953,8 @@ def train_gpt(language, num_epochs, batch_size, grad_acumm, train_csv, eval_csv,
         lr_scheduler=lr_scheduler,
         # it was adjusted accordly for the new step scheme
         lr_scheduler_params=lr_scheduler_params,
+        test_sentences=[],
+    )
     progress(0, desc="Model is currently training. See console for more information")
     # init the model from config
     model = GPTTrainer.init_from_config(config)
@@ -2381,4 +2384,14 @@ if __name__ == "__main__":
                 fn=delete_voice_sample_contents,
                 outputs=[final_progress_data],
             )
-            model_to_train.change(basemodel_or_finetunedmodel_choice, model_to_train, None)
\ No newline at end of file
+            model_to_train.change(basemodel_or_finetunedmodel_choice, model_to_train, None)
+
+    demo.queue().launch(
+        show_api=False,
+        inbrowser=True,
+        share=False,
+        debug=False,
+        server_port=7052,
+        server_name="127.0.0.1",
+    )
+