CLIP hijack rework

This commit is contained in:
AUTOMATIC
2023-01-07 01:45:28 +03:00
parent 3246a2d6b8
commit 79e39fae61
5 changed files with 246 additions and 172 deletions

View File

@@ -79,7 +79,6 @@ class EmbeddingDatabase:
self.word_embeddings[embedding.name] = embedding
# TODO changing between clip and open clip changes tokenization, which will cause embeddings to stop working
ids = model.cond_stage_model.tokenize([embedding.name])[0]
first_id = ids[0]