diff --git a/pyproject.toml b/pyproject.toml index 1d22de6..0ef2174 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -14,7 +14,7 @@ packages = [ name = "rerankers" -version = "0.1.0" +version = "0.1.0post1" description = "A unified API for various document re-ranking models." @@ -52,7 +52,7 @@ dependencies = [ ] [project.optional-dependencies] -all = ["transformers", "torch", "litellm", "requests", "sentencepiece", "protobuf", "colbert-ai"] +all = ["transformers", "torch", "litellm", "requests", "sentencepiece", "protobuf"] transformers = ["transformers", "torch", "sentencepiece", "protobuf"] api = ["requests"] gpt = ["litellm"] diff --git a/rerankers/__init__.py b/rerankers/__init__.py index 76f53b7..cf91382 100644 --- a/rerankers/__init__.py +++ b/rerankers/__init__.py @@ -1,4 +1,4 @@ from rerankers.reranker import Reranker __all__ = ["Reranker"] -__version__ = "0.1.0" +__version__ = "0.1.0post1"