print list of embeddings on reload
This commit is contained in:
parent
b99d3cf6dd
commit
62edfae257
1 changed files with 1 additions and 0 deletions
|
@ -137,6 +137,7 @@ class EmbeddingDatabase:
|
||||||
continue
|
continue
|
||||||
|
|
||||||
print(f"Loaded a total of {len(self.word_embeddings)} textual inversion embeddings.")
|
print(f"Loaded a total of {len(self.word_embeddings)} textual inversion embeddings.")
|
||||||
|
print("Embeddings:", ', '.join(self.word_embeddings.keys()))
|
||||||
|
|
||||||
def find_embedding_at_position(self, tokens, offset):
|
def find_embedding_at_position(self, tokens, offset):
|
||||||
token = tokens[offset]
|
token = tokens[offset]
|
||||||
|
|
Loading…
Reference in a new issue