Update tools/Fooocus/modules/hack_async_worker_class.py

tools/Fooocus/modules/hack_async_worker_class.py

@@ -16,7 +16,7 @@ import modules.core as core
 import modules.flags as flags
 import modules.config
 import modules.patch
-import ldm_patched.modules.model_management
+# import ldm_patched.modules.model_management
 import extras.preprocessors as preprocessors
 import modules.inpaint_worker as inpaint_worker
 import modules.constants as constants
@@ -859,8 +859,8 @@ class Hack_worker():
             execution_start_time = time.perf_counter()
 
             try:
-                if async_task.last_stop is not False:
-                    ldm_patched.modules.model_management.interrupt_current_processing()
+                # if async_task.last_stop is not False:
+                    # ldm_patched.modules.model_management.interrupt_current_processing()
                 positive_cond, negative_cond = task['c'], task['uc']
 
                 if 'cn' in goals:
@@ -903,14 +903,14 @@ class Hack_worker():
                     self.progressbar(async_task, current_progress, 'Checking for NSFW content ...')
                     imgs = default_censor(imgs)
                 outputs.append(imgs[0])
-            except ldm_patched.modules.model_management.InterruptProcessingException as e:
-                if async_task.last_stop == 'skip':
-                    print('User skipped')
-                    async_task.last_stop = False
-                    continue
-                else:
-                    print('User stopped')
-                    break
+            # except ldm_patched.modules.model_management.InterruptProcessingException as e:
+            #     if async_task.last_stop == 'skip':
+            #         print('User skipped')
+            #         async_task.last_stop = False
+            #         continue
+            #     else:
+            #         print('User stopped')
+            #         break
             execution_time = time.perf_counter() - execution_start_time
             print(f'Generating and saving time: {execution_time:.2f} seconds')
         async_task.processing = False