From b124797949a0df56957a8ec32ead8f79f22ed5ca Mon Sep 17 00:00:00 2001 From: kingbri Date: Fri, 2 Aug 2024 14:35:58 -0400 Subject: [PATCH] Dependencies: Re-add sentence-transformers This is actually required for infinity to load a model. Signed-off-by: kingbri --- pyproject.toml | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index 6016864..e53d887 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -48,7 +48,8 @@ dependencies = [ extras = [ # Heavy dependencies that aren't for everyday use "outlines", - "infinity-emb" + "infinity-emb", + "sentence-transformers", ] dev = [ "ruff == 0.3.2"