Spaces:
Running
Running
File size: 2,430 Bytes
9ff79dc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 |
[project]
name = 'colpali_engine'
# dynamic = ["version"]
version = '0.0.1'
description = 'This repository centralizes ressources for the ColPali project.'
authors = [
{ name = 'Manuel Faysse', email = 'manuel.faysse@illuin.tech' },
{ name = 'Hugues Sibille', email = 'hugues.sibille@illuin.tech' },
{ name = 'Tony Wu', email = 'tony.wu@illuin.tech' },
]
readme = 'README.md'
requires-python = '>=3.9'
classifiers = [
'Intended Audience :: Science/Research',
'Intended Audience :: Developers',
'Operating System :: OS Independent',
'Private :: Do Not Upload',
'Programming Language :: Python :: 3',
'Programming Language :: Python :: 3.10',
'Topic :: Scientific/Engineering :: Artificial Intelligence',
'Typing :: Typed',
]
dependencies = [
"torch>=2.2.0",
"transformers>=4.41.1",
"mteb>=1.12.22",
"requests",
"GPUtil",
"peft>=0.11.0, <0.12.0",
]
[project.optional-dependencies]
dev = [
"black>=24.4.2",
"coverage>=7.5.2",
"ipykernel>=6.29.4",
"mypy>=1.10.0",
"pytest>=8.2.1",
"ruff>=0.4.5",
]
train = [
"accelerate==0.30.1",
"configue==5.0.0",
"datasets==2.19.1",
"typer==0.12.3",
"bitsandbytes",
]
[project.urls]
homepage = "https://github.com/ManuelFay/colpali"
[build-system]
requires = ['setuptools', 'setuptools_scm[toml]', 'wheel']
build-backend = 'setuptools.build_meta'
[tool.setuptools_scm]
fallback_version = '0.0.0-dev'
[tool.setuptools]
zip-safe = false
platforms = ['any']
[tool.setuptools.packages.find]
include = ['colpali_engine', 'colpali_engine.*']
[tool.mypy]
check_untyped_defs = true
disallow_untyped_defs = true
enable_error_code = ['ignore-without-code']
exclude = ['docs/']
mypy_path = '$MYPY_CONFIG_FILE_DIR/typings'
no_implicit_optional = true
show_error_codes = true
warn_redundant_casts = true
warn_return_any = true
warn_unused_configs = true
warn_unused_ignores = true
warn_unreachable = true
[[tool.mypy.overrides]]
module = ['transformers', 'transformers.*', 'torch', 'torch.*']
ignore_missing_imports = true
[tool.coverage.run]
include = []
[tool.coverage.report]
exclude_lines = [
'pragma: no cover',
'raise NotImplementedError',
'if __name__ == "__main__":',
'if TYPE_CHECKING:',
'def __repr__',
]
[tool.black]
line-length = 120
[tool.ruff]
select = ["E", "F", "W", "I", "N"]
line-length = 120
[tool.ruff.per-file-ignores]
'__init__.py' = ["F401"]
|