Heng2004 commited on
Commit
255887a
·
verified ·
1 Parent(s): a021341

Update model_utils.py

Browse files
Files changed (1) hide show
  1. model_utils.py +2 -0
model_utils.py CHANGED
@@ -35,6 +35,8 @@ embed_model = SentenceTransformer(EMBED_MODEL_NAME)
35
  embed_model = embed_model.to(device)
36
 
37
 
 
 
38
  def _build_entry_embeddings() -> None:
39
  """
40
  Build embeddings for each textbook entry using title + summary + text
 
35
  embed_model = embed_model.to(device)
36
 
37
 
38
+ # NOTE: called once after load_curriculum() to precompute embeddings.
39
+ # If you ever reload curriculum at runtime, call _build_entry_embeddings() again.
40
  def _build_entry_embeddings() -> None:
41
  """
42
  Build embeddings for each textbook entry using title + summary + text