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

1 files changed, 3 insertions, 6 deletions

diff --git a/modules/textual_inversion/textual_inversion.py b/modules/textual_inversion/textual_inversion.py
index d489ed1e..a040a988 100644
--- a/modules/textual_inversion/textual_inversion.py
+++ b/modules/textual_inversion/textual_inversion.py
@@ -1,6 +1,4 @@
 import os

-import sys

-import traceback

 from collections import namedtuple

 

 import torch

@@ -16,6 +14,7 @@ from torch.utils.tensorboard import SummaryWriter
 

 from modules import shared, devices, sd_hijack, processing, sd_models, images, sd_samplers, sd_hijack_checkpoint

 import modules.textual_inversion.dataset

+from modules.errors import print_error

 from modules.textual_inversion.learn_schedule import LearnRateScheduler

 

 from modules.textual_inversion.image_embedding import embedding_to_b64, embedding_from_b64, insert_image_data_embed, extract_image_data_embed, caption_image_overlay

@@ -207,8 +206,7 @@ class EmbeddingDatabase:
 

                     self.load_from_file(fullfn, fn)

                 except Exception:

-                    print(f"Error loading embedding {fn}:", file=sys.stderr)

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

+                    print_error(f"Error loading embedding {fn}", exc_info=True)

                     continue

 

     def load_textual_inversion_embeddings(self, force_reload=False):

@@ -632,8 +630,7 @@ Last saved image: {html.escape(last_saved_image)}<br/>
         filename = os.path.join(shared.cmd_opts.embeddings_dir, f'{embedding_name}.pt')

         save_embedding(embedding, optimizer, checkpoint, embedding_name, filename, remove_cached_checksum=True)

     except Exception:

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

-        pass

+        print_error("Error training embedding", exc_info=True)

     finally:

         pbar.leave = False

         pbar.close()