sgorblex/M3DocRAG
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

more bearable pyproject.toml

Browse Source
This commit is contained in:
Alessandro Clerici
2025-07-09 15:40:27 +00:00
parent 2ab91099f2
commit ab46556a0a
1 changed files with 18 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
pyproject.toml
View File

@ -1,5 +1,6 @@
[build-system] [build-system]
requires = ["setuptools>=69.5"] # requires = ["setuptools>=69.5"]
requires = ["setuptools"]
build-backend = "setuptools.build_meta" build-backend = "setuptools.build_meta"
[tool.setuptools.packages.find] [tool.setuptools.packages.find]
@ -13,20 +14,23 @@ readme = "README.md"
requires-python = ">=3.10" requires-python = ">=3.10"
classifiers = ["Programming Language :: Python :: 3"] classifiers = ["Programming Language :: Python :: 3"]
dependencies = [ dependencies = [
"accelerate==1.1.0", # "accelerate==1.1.0",
"accelerate",
"loguru", "loguru",
"requests", "requests",
"setuptools==69.5", # "setuptools>=69.5",
"setuptools",
"transformers", "transformers",
"tokenizers", "tokenizers",
"flash-attn==2.5.8", # "flash-attn==2.5.8",
"bitsandbytes==0.43.1", "flash-attn",
# "bitsandbytes==0.43.1",
"bitsandbytes",
"safetensors", "safetensors",
"gpustat", "gpustat",
"icecream", "icecream",
"pdf2image", "pdf2image",
"numpy==1.26.4", "numpy",
"torchvision",
"jsonlines", "jsonlines",
"editdistance", "editdistance",
"einops", "einops",
@ -34,13 +38,17 @@ dependencies = [
"peft", "peft",
"timm", "timm",
"sentencepiece", "sentencepiece",
"colpali-engine==0.3.1", # "colpali-engine==0.3.1",
"colpali-engine",
"easyocr", "easyocr",
"qwen-vl-utils", "qwen-vl-utils",
"faiss-cpu", "faiss-cpu",
"word2number", "word2number",
"datasets>=3.0.0", # "datasets>=3.0.0",
"python-dotenv", "python-dotenv",
# "torchvision==0.22.1+cu128",
"torchvision",
"datasets>=3.6.0",
] ]
[tool.ruff] [tool.ruff]