diff options
Diffstat (limited to 'infer.py')
-rw-r--r-- | infer.py | 1 |
1 files changed, 1 insertions, 0 deletions
@@ -235,6 +235,7 @@ def load_embeddings(pipeline, embeddings_dir): | |||
235 | pipeline.text_encoder.text_model.embeddings, | 235 | pipeline.text_encoder.text_model.embeddings, |
236 | Path(embeddings_dir) | 236 | Path(embeddings_dir) |
237 | ) | 237 | ) |
238 | pipeline.text_encoder.text_model.embeddings.persist() | ||
238 | print(f"Added {len(added_tokens)} tokens from embeddings dir: {list(zip(added_tokens, added_ids))}") | 239 | print(f"Added {len(added_tokens)} tokens from embeddings dir: {list(zip(added_tokens, added_ids))}") |
239 | 240 | ||
240 | 241 | ||