refiners/pyproject.toml

71 lines
2.3 KiB
TOML
Raw Normal View History

2023-08-04 13:28:41 +00:00
[tool.poetry]
name = "refiners"
2023-10-20 16:04:55 +00:00
version = "0.2.0"
2023-08-04 13:28:41 +00:00
description = "The simplest way to train and run adapters on top of foundational models"
authors = [
"The Finegrain Team <bonjour@lagon.tech>",
]
license = "MIT"
readme = "README.md"
packages = [{include = "refiners", from = "src"}]
[tool.poetry.dependencies]
python = ">=3.10,<3.12"
jaxtyping = "^0.2.14"
2023-10-10 09:50:27 +00:00
torch = "^2.1.0"
2023-08-04 13:28:41 +00:00
safetensors = "^0.3.0"
numpy = "^1.24.2"
2023-09-21 11:36:44 +00:00
pillow = ">=9.5.0"
2023-08-04 13:28:41 +00:00
datasets = {version = "^2.14.0", optional = true}
tomli = {version = "^2.0.1", optional = true}
wandb = {version = "^0.15.7", optional = true}
loguru = {version = "^0.7.0", optional = true}
bitsandbytes = {version = "^0.41.0", optional = true}
prodigyopt = {version = "^1.0", optional = true}
2023-08-04 17:40:50 +00:00
pydantic = {version = "~2.0.3", optional = true}
# Added scipy as a work around until this PR gets merged:
# https://github.com/TimDettmers/bitsandbytes/pull/525
2023-08-04 13:28:41 +00:00
scipy = {version = "^1.11.1", optional = true}
torchvision = {version = "^0.16.0", optional = true}
diffusers = {version = "^0.21.0", optional = true}
2023-09-13 13:15:50 +00:00
transformers = {version = "^4.27.4", optional = true}
piq = {version = "^0.7.1", optional = true}
invisible-watermark = {version = "^0.2.0", optional = true}
# An unofficial Python package for Meta AI's Segment Anything Model:
# https://github.com/opengeos/segment-anything
segment-anything-py = {version = "1.0", optional = true}
2023-08-04 13:28:41 +00:00
[tool.poetry.extras]
training = ["datasets", "tomli", "wandb", "loguru", "bitsandbytes", "prodigyopt", "pydantic", "scipy", "torchvision"]
conversion = ["diffusers", "transformers", "segment-anything-py"]
test = ["diffusers", "transformers", "piq", "invisible-watermark", "segment-anything-py", "torchvision"]
2023-08-04 13:28:41 +00:00
[tool.poetry.group.dev.dependencies]
black = "^23.1.0"
pytest = "^7.2.2"
isort = "^5.12.0"
ipykernel = "^6.22.0"
pyright = "^1.1.318"
ruff = "^0.0.281"
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"
[tool.black]
line-length = 120
[tool.ruff]
ignore = [
"F722", # forward-annotation-syntax-error, because of Jaxtyping
"E731", # do-not-assign-lambda
"E501", # line-too-long, because Black (https://beta.ruff.rs/docs/faq/#is-ruff-compatible-with-black)
]
line-length = 120
[tool.pyright]
include = ["src/refiners", "tests", "scripts"]
2023-08-16 09:17:55 +00:00
strict = ["*"]
2023-08-04 13:28:41 +00:00
exclude = ["**/__pycache__"]
reportMissingTypeStubs = "warning"