wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-20 03:10:16 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

Search recursively in subfolders for embeddings.

Browse Source
This commit is contained in:
comfyanonymous 2023-05-05 01:28:48 -04:00
parent 93fc8c1ea9
commit a6d2a9487d
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
comfy/sd1_clip.py
View File

@ -191,11 +191,20 @@ def safe_load_embed_zip(embed_path):
del embed del embed
return out return out
def expand_directory_list(directories):
dirs = set()
for x in directories:
dirs.add(x)
for root, subdir, file in os.walk(x, followlinks=True):
dirs.add(root)
return list(dirs)
def load_embed(embedding_name, embedding_directory): def load_embed(embedding_name, embedding_directory):
if isinstance(embedding_directory, str): if isinstance(embedding_directory, str):
embedding_directory = [embedding_directory] embedding_directory = [embedding_directory]
embedding_directory = expand_directory_list(embedding_directory)
valid_file = None valid_file = None
for embed_dir in embedding_directory: for embed_dir in embedding_directory:
embed_path = os.path.join(embed_dir, embedding_name) embed_path = os.path.join(embed_dir, embedding_name)