diff --git a/pyproject.toml b/pyproject.toml index 1ad9edb..39dffce 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -6,7 +6,7 @@ readme = "README.md" license = {text = "MIT"} requires-python = ">=3.10" dependencies = [ - "torch>=2.0", + "torch>=2.14.1", ] [project.optional-dependencies] diff --git a/requirements.txt b/requirements.txt index 470fed2..e583ba6 100644 --- a/requirements.txt +++ b/requirements.txt @@ -5,7 +5,7 @@ # Runtime-only (ONNX deployment, zero torch): pip install onnxruntime tokenizers numpy # ── Training (GPU) ── -torch>=2.0 # 2.14.0+cu126 tested; CPU-only works for inference +torch>=2.14.1 # 2.14.0+cu126 tested; CPU-only works for inference # ── Embedding backbone (frozen, inference-only) ── transformers>=4.40 # 5.18.0 tested; loads bge-m3 / e5-small / any HF model