diff --git a/poetry.lock b/poetry.lock index 9b5761a539e4cc8613d27a609524681a337008ad..05ee414c8f606bffaebe4133ee5b1ab158136944 100644 --- a/poetry.lock +++ b/poetry.lock @@ -4218,7 +4218,7 @@ testing = ["big-O", "jaraco.functools", "jaraco.itertools", "jaraco.test", "more bedrock = ["boto3"] google = ["google-cloud-aiplatform"] hybrid = ["pinecone-text"] -local = ["llama-cpp-python", "tokenizers", "torch", "transformers"] +local = ["huggingface-hub", "llama-cpp-python", "tokenizers", "torch", "transformers"] mistralai = ["mistralai"] pinecone = ["pinecone-client"] processing = ["matplotlib"] @@ -4228,4 +4228,4 @@ vision = ["pillow", "torch", "torchvision", "transformers"] [metadata] lock-version = "2.0" python-versions = ">=3.9,<3.13" -content-hash = "ea02ac49c6a0107091cc76153249bece31def22b7d16758c9c024b78d47a9418" +content-hash = "7a611e25c44d7d5b997d0826b2c96a4e0479eb343171aecae0efeee9fee15728" diff --git a/pyproject.toml b/pyproject.toml index dbf4785fc6f87e4760b117252cc8927ffef39bc1..cd0e811aac94e5a02e8df4d37564f46bfbd91706 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -28,6 +28,7 @@ pinecone-text = {version = ">=0.7.1,<0.8.0", optional = true} torch = {version = ">=2.1.0,<2.6.0", optional = true} transformers = {version = ">=4.36.2", optional = true} tokenizers = {version = ">=0.19", optional = true} +huggingface-hub = {version = "!=0.23.1", optional = true} llama-cpp-python = {version = "^0.2.28", optional = true} colorama = "^0.4.6" pinecone-client = {version=">=3.0.0,<4.0.0", optional = true} @@ -43,7 +44,7 @@ boto3 = { version = "^1.34.98", optional = true } [tool.poetry.extras] hybrid = ["pinecone-text"] -local = ["torch", "transformers", "tokenizers", "llama-cpp-python"] +local = ["torch", "transformers", "tokenizers", "huggingface-hub", "llama-cpp-python"] pinecone = ["pinecone-client"] vision = ["torch", "torchvision", "transformers", "pillow"] processing = ["matplotlib"]