Unverified Commit 143ed5a4 authored by AUTOMATIC1111's avatar AUTOMATIC1111 Committed by GitHub
Browse files

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

allow loading embeddings from subdirectories
parents 8a13afd2 81133d41
Loading
Loading
Loading
Loading
+12 −11
Original line number Diff line number Diff line
@@ -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