Update flux_train_ui.py

flux_train_ui.py

@@ -243,6 +243,9 @@ def start_training(
     with open(config_path, "w") as f:
         yaml.dump(config, f)
     
+    
+    
+    print(f"[INFO] Starting training with config: {config_path}")
     # run the job locally
     job = get_job(config_path)
     job.run()
@@ -250,6 +253,21 @@ def start_training(
 
     return f"Training completed successfully. Model saved as {slugged_lora_name}"
 
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
 config_yaml = '''
 device: cuda:0
 model:
@@ -301,6 +319,15 @@ h3{margin-top: 0}
 .tabitem{border: 0px}
 .group_padding{padding: .55em}
 """
+
+
+
+
+
+
+
+
+
 with gr.Blocks(theme=theme, css=css) as demo:
     gr.Markdown(
         """# LoRA Ease for FLUX 🧞‍♂️