diff --git a/base_embedding.py b/llmengine/base_embedding.py similarity index 100% rename from base_embedding.py rename to llmengine/base_embedding.py diff --git a/base_reranker.py b/llmengine/base_reranker.py similarity index 100% rename from base_reranker.py rename to llmengine/base_reranker.py diff --git a/qwen3_reranker.py b/llmengine/qwen3_reranker.py similarity index 100% rename from qwen3_reranker.py rename to llmengine/qwen3_reranker.py diff --git a/qwen3embedding.py b/llmengine/qwen3embedding.py similarity index 100% rename from qwen3embedding.py rename to llmengine/qwen3embedding.py diff --git a/pyproject.toml b/pyproject.toml index f2df66d..73d37f7 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -9,6 +9,8 @@ license = {text = "MIT"} dependencies = [ "torch", "transformers", + "sentence-transformers>=2.7.0", + # "flash_attention_2", "mistral-common", "accelerate" ]