catch exception for all paste_fields callable

IllusiveMan196 · Apr 9, 2024 · ef83f68 · ef83f68
1 parent 600f339
commit ef83f68
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/modules/infotext_utils.py b/modules/infotext_utils.py
@@ -8,7 +8,7 @@
 
 import gradio as gr
 from modules.paths import data_path
-from modules import shared, ui_tempdir, script_callbacks, processing, infotext_versions, images, prompt_parser
+from modules import shared, ui_tempdir, script_callbacks, processing, infotext_versions, images, prompt_parser, errors
 from PIL import Image
 
 sys.modules['modules.generation_parameters_copypaste'] = sys.modules[__name__]  # alias for old name
@@ -488,7 +488,11 @@ def paste_func(prompt):
 
         for output, key in paste_fields:
             if callable(key):
-                v = key(params)
+                try:
+                    v = key(params)
+                except Exception:
+                    errors.report(f"Error executing {key}", exc_info=True)
+                    v = None
             else:
                 v = params.get(key, None)