From 1d9845557f90646265e4cea4eaa89caed7429b18 Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Mon, 20 Feb 2023 15:18:55 -0500 Subject: [PATCH] reduced verbosity of embed loading messages --- ldm/generate.py | 3 ++- ldm/modules/textual_inversion_manager.py | 1 - 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/ldm/generate.py b/ldm/generate.py index d7ea87a5fd..76629ff2f9 100644 --- a/ldm/generate.py +++ b/ldm/generate.py @@ -964,6 +964,7 @@ class Generate: seed_everything(random.randrange(0, np.iinfo(np.uint32).max)) if self.embedding_path is not None: + print(f'>> Loading embeddings from {self.embedding_path}') for root, _, files in os.walk(self.embedding_path): for name in files: ti_path = os.path.join(root, name) @@ -971,7 +972,7 @@ class Generate: ti_path, defer_injecting_tokens=True ) print( - f'>> Textual inversions available: {", ".join(self.model.textual_inversion_manager.get_all_trigger_strings())}' + f'>> Textual inversion triggers: {", ".join(self.model.textual_inversion_manager.get_all_trigger_strings())}' ) self.model_name = model_name diff --git a/ldm/modules/textual_inversion_manager.py b/ldm/modules/textual_inversion_manager.py index 3a29aa5702..bf0d3ed8b9 100644 --- a/ldm/modules/textual_inversion_manager.py +++ b/ldm/modules/textual_inversion_manager.py @@ -69,7 +69,6 @@ class TextualInversionManager: return try: - print(f'>> Scanning {str(ckpt_path)} for embedding terms') scan_result = scan_file_path(str(ckpt_path)) if scan_result.infected_files == 1: print(