From 4b859e235b01d208d3e7aa9e6f939eeca5633a90 Mon Sep 17 00:00:00 2001 From: yumoqing Date: Thu, 19 Jun 2025 17:31:10 +0800 Subject: [PATCH] bugfix --- base_embedding.py => llmengine/base_embedding.py | 0 base_reranker.py => llmengine/base_reranker.py | 0 qwen3_reranker.py => llmengine/qwen3_reranker.py | 0 qwen3embedding.py => llmengine/qwen3embedding.py | 0 pyproject.toml | 2 ++ 5 files changed, 2 insertions(+) rename base_embedding.py => llmengine/base_embedding.py (100%) rename base_reranker.py => llmengine/base_reranker.py (100%) rename qwen3_reranker.py => llmengine/qwen3_reranker.py (100%) rename qwen3embedding.py => llmengine/qwen3embedding.py (100%) 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" ]