Skip to content

Commit 16b9883

Browse files
committed
Added thread.lock for embedder init
1 parent 56b83b3 commit 16b9883

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

tasks/update_embeddings.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
os.environ.setdefault("OMP_NUM_THREADS", "1")
2020
os.environ.setdefault("TOKENIZERS_PARALLELISM", "false")
2121

22+
MODEL_INIT_LOCK = threading.Lock()
2223

2324
"""
2425
High-level call order (condensed):
@@ -211,7 +212,11 @@ def _get_embedder(
211212
"""
212213
cached = getattr(thread_local, "embedder", None)
213214
if cached is None:
214-
thread_local.embedder = EmbedderTools(model_name=model_name, **embedder_init_kwargs)
215+
with MODEL_INIT_LOCK:
216+
thread_local.embedder = EmbedderTools(
217+
model_name=model_name,
218+
**embedder_init_kwargs,
219+
)
215220
cached = thread_local.embedder
216221
return cached
217222

0 commit comments

Comments
 (0)