Fix script mode

Fixes #443

Author: semjon00

scripts/depthmap.py

@@ -55,7 +55,7 @@ def run(self, p, *inputs):
             # TODO: could reuse
             if hasattr(processed, 'all_prompts') and shared.opts.enable_pnginfo:
                 info = create_infotext(
-                    processed, processed.all_prompts, processed.all_seeds, processed.all_subseeds, "", 0, input_i)
+                    p, processed.all_prompts, processed.all_seeds, processed.all_subseeds, "", 0, input_i)
             else:
                 info = None
 

src/common_ui.py

@@ -437,6 +437,9 @@ def format_exception(e: Exception):
     elif "torch.hub.load('facebookresearch/dinov2'," in traceback.format_exc():
         msg += ('<h4>To use Depth Anything integration in WebUI mode, please add "--disable-safe-unpickle" to the command line flags. '
                 'Alternatively, use Standalone mode. This is a known issue.')
+    elif "Error(s) in loading state_dict " in traceback.format_exc():
+        msg += ('<h4>There was issue during loading the model.'
+                'Please add "--disable-safe-unpickle" to the command line flags. This is a known issue.')
     elif 'out of GPU memory' not in msg:
         msg += \
             'Please report this issue ' \