From 102a271410e893432e7f1e2476e2dde7c51dd40c Mon Sep 17 00:00:00 2001 From: Karl Higley Date: Wed, 4 Dec 2024 12:19:18 -0500 Subject: [PATCH] Fix paths to the safetensors files --- src/poprox_recommender/recommenders.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/poprox_recommender/recommenders.py b/src/poprox_recommender/recommenders.py index 4b9ea55a..64e31f8b 100644 --- a/src/poprox_recommender/recommenders.py +++ b/src/poprox_recommender/recommenders.py @@ -85,8 +85,8 @@ def build_pipelines(num_slots: int, device: str) -> dict[str, Pipeline]: num_slots: The number of items to recommend. """ - article_embedder = NRMSArticleEmbedder(model_file_path("news_encoder.safetensors"), device) - user_embedder = NRMSUserEmbedder(model_file_path("user_encoder.safetensors"), device) + article_embedder = NRMSArticleEmbedder(model_file_path("nrms-mind/news_encoder.safetensors"), device) + user_embedder = NRMSUserEmbedder(model_file_path("nrms-mind/user_encoder.safetensors"), device) topk_ranker = TopkRanker(num_slots=num_slots) mmr = MMRDiversifier(num_slots=num_slots)