Add & use modules.errors.print_error where currently printing exception info by hand

1 files changed, 2 insertions, 3 deletions

diff --git a/modules/interrogate.py b/modules/interrogate.py
index 111b1322..d36e1a5a 100644
--- a/modules/interrogate.py
+++ b/modules/interrogate.py
@@ -1,6 +1,5 @@
 import os

 import sys

-import traceback

 from collections import namedtuple

 from pathlib import Path

 import re

@@ -12,6 +11,7 @@ from torchvision import transforms
 from torchvision.transforms.functional import InterpolationMode

 

 from modules import devices, paths, shared, lowvram, modelloader, errors

+from modules.errors import print_error

 

 blip_image_eval_size = 384

 clip_model_name = 'ViT-L/14'

@@ -216,8 +216,7 @@ class InterrogateModels:
                             res += f", {match}"

 

         except Exception:

-            print("Error interrogating", file=sys.stderr)

-            print(traceback.format_exc(), file=sys.stderr)

+            print_error("Error interrogating", exc_info=True)

             res += "<error>"

 

         self.unload()