2023-12-08 11:26:50 +00:00
|
|
|
[project]
|
2023-08-04 13:28:41 +00:00
|
|
|
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"
|
2023-12-08 11:26:50 +00:00
|
|
|
authors = [{ name = "The Finegrain Team", email = "bonjour@lagon.tech" }]
|
2023-08-04 13:28:41 +00:00
|
|
|
license = "MIT"
|
2023-12-08 11:26:50 +00:00
|
|
|
dependencies = [
|
|
|
|
"torch>=2.1.1",
|
|
|
|
"safetensors>=0.4.0",
|
|
|
|
"pillow>=10.1.0",
|
|
|
|
"jaxtyping>=0.2.23",
|
2023-12-08 17:55:59 +00:00
|
|
|
"packaging>=23.2",
|
2023-12-08 11:26:50 +00:00
|
|
|
]
|
2023-08-04 13:28:41 +00:00
|
|
|
readme = "README.md"
|
2023-12-08 11:26:50 +00:00
|
|
|
requires-python = ">= 3.10"
|
2023-08-04 13:28:41 +00:00
|
|
|
|
2023-12-08 11:26:50 +00:00
|
|
|
[project.optional-dependencies]
|
|
|
|
training = [
|
|
|
|
"bitsandbytes>=0.41.2.post2",
|
|
|
|
"pydantic>=2.5.2",
|
|
|
|
"prodigyopt>=1.0",
|
|
|
|
"torchvision>=0.16.1",
|
|
|
|
"loguru>=0.7.2",
|
|
|
|
"wandb>=0.16.0",
|
|
|
|
# Added scipy as a work around until this PR gets merged:
|
|
|
|
# https://github.com/TimDettmers/bitsandbytes/pull/525
|
|
|
|
"scipy>=1.11.4",
|
|
|
|
"datasets>=2.15.0",
|
2023-12-15 17:02:32 +00:00
|
|
|
"tomli>=2.0.1",
|
2023-12-08 11:26:50 +00:00
|
|
|
]
|
|
|
|
test = [
|
|
|
|
"diffusers>=0.24.0",
|
|
|
|
"transformers>=4.35.2",
|
|
|
|
"piq>=0.8.0",
|
|
|
|
"invisible-watermark>=0.2.0",
|
|
|
|
"torchvision>=0.16.1",
|
|
|
|
# An unofficial Python package for Meta AI's Segment Anything Model:
|
|
|
|
# https://github.com/opengeos/segment-anything
|
|
|
|
"segment-anything-py>=1.0",
|
|
|
|
]
|
|
|
|
conversion = [
|
|
|
|
"diffusers>=0.24.0",
|
|
|
|
"transformers>=4.35.2",
|
|
|
|
"segment-anything-py>=1.0",
|
2023-12-14 03:53:06 +00:00
|
|
|
"requests>=2.26.0",
|
|
|
|
"tqdm>=4.62.3",
|
2023-12-08 11:26:50 +00:00
|
|
|
]
|
2023-08-04 13:28:41 +00:00
|
|
|
|
2023-12-08 11:26:50 +00:00
|
|
|
[build-system]
|
|
|
|
requires = ["hatchling"]
|
|
|
|
build-backend = "hatchling.build"
|
2023-08-04 13:28:41 +00:00
|
|
|
|
|
|
|
|
2023-12-08 11:26:50 +00:00
|
|
|
[tool.rye]
|
|
|
|
managed = true
|
|
|
|
dev-dependencies = [
|
2023-12-29 09:59:51 +00:00
|
|
|
"pyright == 1.1.342",
|
2023-12-08 11:26:50 +00:00
|
|
|
"ruff>=0.0.292",
|
|
|
|
"docformatter>=1.7.5",
|
|
|
|
"pytest>=7.4.2",
|
2024-01-08 15:18:11 +00:00
|
|
|
"mkdocs-material>=9.5.3",
|
2023-12-08 11:26:50 +00:00
|
|
|
]
|
|
|
|
|
|
|
|
|
|
|
|
[tool.hatch.metadata]
|
|
|
|
allow-direct-references = true
|
|
|
|
|
|
|
|
[tool.rye.scripts]
|
|
|
|
lint = { chain = ["ruff format .", "ruff --fix ."] }
|
2024-01-08 15:54:03 +00:00
|
|
|
serve-docs = "mkdocs serve"
|
2023-08-04 13:28:41 +00:00
|
|
|
|
|
|
|
[tool.black]
|
|
|
|
line-length = 120
|
|
|
|
|
|
|
|
[tool.ruff]
|
2023-12-11 10:40:27 +00:00
|
|
|
src = ["src"] # see https://docs.astral.sh/ruff/settings/#src
|
|
|
|
select = [
|
|
|
|
"I", # isort
|
|
|
|
]
|
2023-08-04 13:28:41 +00:00
|
|
|
ignore = [
|
2023-12-08 11:26:50 +00:00
|
|
|
"F722", # forward-annotation-syntax-error, because of Jaxtyping
|
|
|
|
"E731", # do-not-assign-lambda
|
2023-08-04 13:28:41 +00:00
|
|
|
]
|
|
|
|
line-length = 120
|
|
|
|
|
2023-12-11 10:40:27 +00:00
|
|
|
[tool.ruff.lint.isort]
|
2023-12-11 12:50:27 +00:00
|
|
|
# Allow this kind of import on a single line:
|
|
|
|
#
|
|
|
|
# from torch import device as Device, dtype as DType
|
|
|
|
#
|
2023-12-11 10:40:27 +00:00
|
|
|
combine-as-imports = true
|
|
|
|
|
2023-12-08 11:26:50 +00:00
|
|
|
[tool.docformatter]
|
|
|
|
black = true
|
|
|
|
|
2023-08-04 13:28:41 +00:00
|
|
|
[tool.pyright]
|
2023-08-15 16:06:50 +00:00
|
|
|
include = ["src/refiners", "tests", "scripts"]
|
2023-08-16 09:17:55 +00:00
|
|
|
strict = ["*"]
|
2024-01-20 17:37:49 +00:00
|
|
|
exclude = ["**/__pycache__", "tests/weights"]
|
2023-08-04 13:28:41 +00:00
|
|
|
reportMissingTypeStubs = "warning"
|