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", + ) +