fix: simplify torch dependencies and update pinned docling deps (#190)

* fix: simplify torch dependencies and update pinned docling deps

Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>

* update docling-ibm-models

Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>

---------

Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>
This commit is contained in:
Michele Dolfi
2024-10-30 18:44:08 +01:00
committed by GitHub
parent 904d24d600
commit eb679ccbb4
2 changed files with 121 additions and 118 deletions

View File

@@ -21,24 +21,13 @@ keywords= ["docling", "convert", "document", "pdf", "layout model", "segmentatio
packages = [{include = "docling"}]
[tool.poetry.dependencies]
##############
# constraints:
##############
torch = [
{version = "^2.2.2", optional = true, markers = "sys_platform != 'darwin' or platform_machine != 'x86_64'"},
{version = "~2.2.2", optional = true, markers = "sys_platform == 'darwin' and platform_machine == 'x86_64'"}
]
torchvision = [
{version = "^0", optional = true, markers = "sys_platform != 'darwin' or platform_machine != 'x86_64'"},
{version = "~0.17.2", optional = true, markers = "sys_platform == 'darwin' and platform_machine == 'x86_64'"}
]
######################
# actual dependencies:
######################
python = "^3.10"
pydantic = "^2.0.0"
docling-core = "^2.2.3"
docling-ibm-models = "^2.0.1"
docling-core = "^2.3.0"
docling-ibm-models = "^2.0.3"
deepsearch-glm = "^0.26.1"
filetype = "^1.2.0"
pypdfium2 = "^4.30.0"
@@ -47,7 +36,7 @@ huggingface_hub = ">=0.23,<1"
requests = "^2.32.3"
easyocr = "^1.7"
tesserocr = { version = "^2.7.1", optional = true }
docling-parse = "^2.0.0"
docling-parse = "^2.0.2"
certifi = ">=2024.7.4"
rtree = "^1.3.0"
scipy = "^1.14.1"
@@ -94,6 +83,19 @@ langchain-huggingface = "^0.0.3"
langchain-milvus = "^0.1.4"
langchain-text-splitters = "^0.2.4"
[tool.poetry.group.mac_intel]
optional = true
[tool.poetry.group.mac_intel.dependencies]
torch = [
{markers = "sys_platform != 'darwin' or platform_machine != 'x86_64'", version = "^2.2.2"},
{markers = "sys_platform == 'darwin' and platform_machine == 'x86_64'", version = "~2.2.2"}
]
torchvision = [
{markers = "sys_platform != 'darwin' or platform_machine != 'x86_64'", version = "^0"},
{markers = "sys_platform == 'darwin' and platform_machine == 'x86_64'", version = "~0.17.2"}
]
[tool.poetry.extras]
tesserocr = ["tesserocr"]