From b3924764c35c9b4bc847c91858d835ce50660827 Mon Sep 17 00:00:00 2001
From: James Briggs <james.briggs@hotmail.com>
Date: Sat, 1 Jun 2024 18:26:45 +0800
Subject: [PATCH] fix: hf hub version limit
---
poetry.lock | 4 ++--
pyproject.toml | 3 ++-
2 files changed, 4 insertions(+), 3 deletions(-)
diff --git a/poetry.lock b/poetry.lock
index 9b5761a5..05ee414c 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 dbf4785f..cd0e811a 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"]
--
GitLab