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}
|
2023-10-11 13:39:36 +00:00
|
|
|
# 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}
|
2023-10-10 09:49:50 +00:00
|
|
|
torchvision = {version = "^0.16.0", optional = true}
|
2023-09-24 14:05:42 +00:00
|
|
|
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}
|
2023-10-21 09:08:01 +00:00
|
|
|
# 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]
|
2023-08-07 10:10:16 +00:00
|
|
|
training = ["datasets", "tomli", "wandb", "loguru", "bitsandbytes", "prodigyopt", "pydantic", "scipy", "torchvision"]
|
2023-10-21 09:08:01 +00:00
|
|
|
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]
|
2023-08-15 16:06:50 +00:00
|
|
|
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"
|
