diff --git a/changai/changai/replicate_model_files/entity_retriever/.cog/tmp/build20251221171616.080360/requirements.txt b/changai/changai/replicate_model_files/entity_retriever/.cog/tmp/build20251221171616.080360/requirements.txt
index 07f72be..2053b6f 100644
--- a/changai/changai/replicate_model_files/entity_retriever/.cog/tmp/build20251221171616.080360/requirements.txt
+++ b/changai/changai/replicate_model_files/entity_retriever/.cog/tmp/build20251221171616.080360/requirements.txt
@@ -1,5 +1,5 @@
 --extra-index-url https://download.pytorch.org/whl/cpu
-torch==2.3.1
+torch==2.7.1+cpu
 faiss-cpu==1.8.0
 numpy==1.26.4
 sentence-transformers>=2.6.0