Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

reduced verbosity of embed loading messages

Browse Source
This commit is contained in:
Lincoln Stein
2023-02-20 15:18:55 -05:00
parent 0d22fd59ed
commit 1d9845557f
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
ldm/generate.py
View File

@ -964,6 +964,7 @@ class Generate:
seed_everything(random.randrange(0, np.iinfo(np.uint32).max)) seed_everything(random.randrange(0, np.iinfo(np.uint32).max))
if self.embedding_path is not None: 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 root, _, files in os.walk(self.embedding_path):
for name in files: for name in files:
ti_path = os.path.join(root, name) ti_path = os.path.join(root, name)
@ -971,7 +972,7 @@ class Generate:
ti_path, defer_injecting_tokens=True ti_path, defer_injecting_tokens=True
) )
print( 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 self.model_name = model_name

1
ldm/modules/textual_inversion_manager.py
View File

@ -69,7 +69,6 @@ class TextualInversionManager:
return return
try: try:
print(f'>> Scanning {str(ckpt_path)} for embedding terms')
scan_result = scan_file_path(str(ckpt_path)) scan_result = scan_file_path(str(ckpt_path))
if scan_result.infected_files == 1: if scan_result.infected_files == 1:
print( print(