Merge pull request #6384 from faber6/loads-ti-from-subdirs

allow loading embeddings from subdirectories
2023-01-06 07:56:48 +03:00 · 2023-01-06 07:56:48 +03:00 · 143ed5a42d
commit 143ed5a42d
parent 8a13afd216 81133d4168
1 changed files with 11 additions and 10 deletions
--- a/modules/textual_inversion/textual_inversion.py
+++ b/modules/textual_inversion/textual_inversion.py
@ -149,9 +149,10 @@ class EmbeddingDatabase:
            else:
                self.skipped_embeddings[name] = embedding

-        for fn in os.listdir(self.embeddings_dir):
+        for root, dirs, fns in os.walk(self.embeddings_dir):
+            for fn in fns:
                try:
-                fullfn = os.path.join(self.embeddings_dir, fn)
+                    fullfn = os.path.join(root, fn)

                    if os.stat(fullfn).st_size == 0:
                        continue