diff --git a/pyproject.toml b/pyproject.toml index e06a94a..bb0f870 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -12,7 +12,7 @@ requires-python = "==3.12.11" dependencies = [ "faiss-cpu==1.8.0", "transformers==4.44.2", - "torch==2.4.0", + "torch==2.8.0", "sentencepiece==0.2.0", "tqdm==4.66.5", "numpy==1.26.4", diff --git a/requirements.txt b/requirements.txt index 27bfce3..2e3fedd 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,6 @@ faiss-cpu==1.8.0 transformers==4.44.2 -torch==2.4.0 +torch==2.8.0 sentencepiece==0.2.0 tqdm==4.66.5 numpy==1.26.4