fixing stop button

Author: neph1

run_trainer.py

@@ -94,10 +94,15 @@ def run(self, config: Config, finetrainers_path: str, log_file: str):
         return "Unknown result"
 
     def stop(self):
-        self.running = False
-        if self.process:
-            self.process.terminate()
-            time.sleep(3)
-            if self.process.poll() is None:
-                self.process.kill()
-        return "Training stopped"
+        try:
+            self.running = False
+            if self.process:
+                self.process.terminate()
+                time.sleep(3)
+                if self.process.poll() is None:
+                    self.process.kill()
+        except Exception as e:
+            return f"Error stopping training: {e}"
+        finally:
+            self.process.wait()
+        return "Training forcibly stopped"