mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2026-01-07 17:20:19 +01:00
extensions/superboogav2: existing embedding check bug fix (#6898)
This commit is contained in:
parent
c71a2af5ab
commit
60ac495d59
|
|
@ -148,7 +148,7 @@ class ChromaCollector():
|
|||
id_ = new_ids[i]
|
||||
metadata = metadatas[i] if metadatas is not None else None
|
||||
embedding = self.embeddings_cache.get(text)
|
||||
if embedding is not None and embedding.any():
|
||||
if embedding is not None and any(embedding):
|
||||
existing_texts.append(text)
|
||||
existing_embeddings.append(embedding)
|
||||
existing_ids.append(id_)
|
||||
|
|
|
|||
Loading…
Reference in a new issue