From a985afd857552a140d1a36fd35e97a3e9a58cd81 Mon Sep 17 00:00:00 2001
From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com>
Date: Tue, 13 Aug 2024 15:45:45 -0700
Subject: [PATCH] thread safety

---
 modules_forge/main_thread.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/modules_forge/main_thread.py b/modules_forge/main_thread.py
index f0ed6cc8..eb3e7889 100644
--- a/modules_forge/main_thread.py
+++ b/modules_forge/main_thread.py
@@ -26,10 +26,10 @@ class Task:
 
     def work(self):
         global last_exception
-        self.exception = None
-        last_exception = None
         try:
             self.result = self.func(*self.args, **self.kwargs)
+            self.exception = None
+            last_exception = None
         except Exception as e:
             traceback.print_exc()
             print(e)