add embedding layer impl

2026-04-26 09:18:59 +00:00 · 2024-08-04 13:10:08 -07:00
parent 21c8608373
commit cb9b155645
2 changed files with 45 additions and 21 deletions
--- a/backend/text_processing/textual_inversion.py
+++ b/backend/text_processing/textual_inversion.py
@@ -116,6 +116,7 @@ class EmbeddingDatabase:
        self.skipped_embeddings = {}
        self.expected_shape = expected_shape
        self.tokenizer = tokenizer
+        self.fixes = []

    def add_embedding_dir(self, path):
        self.embedding_dirs[path] = DirWithTextualInversionEmbeddings(path)