|
[project] |
|
name = "refiners" |
|
dynamic = ["version"] |
|
description = "The simplest way to train and run adapters on top of foundation models" |
|
authors = [{ name = "The Finegrain Team", email = "[email protected]" }] |
|
license = { text = "MIT License" } |
|
dependencies = [ |
|
"torch>=2.1.1", |
|
"safetensors>=0.4.5", |
|
"pillow>=10.4.0", |
|
"jaxtyping>=0.2.23", |
|
"packaging>=23.2", |
|
"numpy>=1.26.4", |
|
] |
|
readme = "README.md" |
|
requires-python = ">= 3.10" |
|
classifiers = [ |
|
"Typing :: Typed", |
|
"Intended Audience :: Developers", |
|
"Intended Audience :: Science/Research", |
|
"Programming Language :: Python :: 3.10", |
|
"Programming Language :: Python :: 3.11", |
|
"Topic :: Software Development :: Libraries :: Python Modules", |
|
"Topic :: Scientific/Engineering :: Artificial Intelligence", |
|
"License :: OSI Approved :: MIT License", |
|
] |
|
|
|
[project.urls] |
|
source = "https://github.com/finegrain-ai/refiners" |
|
documentation = "https://refine.rs/" |
|
|
|
[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", |
|
"neptune>=1.10.4", |
|
"datasets>=2.15.0", |
|
"tomli>=2.0.1", |
|
"gitpython>=3.1.43", |
|
] |
|
test = [ |
|
"diffusers>=0.26.1", |
|
"transformers>=4.35.2", |
|
"piq>=0.8.0", |
|
"torchvision>=0.16.1", |
|
|
|
|
|
"segment-anything-py>=1.0", |
|
|
|
"segment-anything-hq>=0.3", |
|
|
|
|
|
"timm>=0.5.0", |
|
"sentencepiece>=0.2.0", |
|
] |
|
conversion = [ |
|
"diffusers>=0.26.1", |
|
"transformers>=4.35.2", |
|
"segment-anything-py>=1.0", |
|
"requests>=2.26.0", |
|
"tqdm>=4.62.3", |
|
"gdown>=5.2.0", |
|
] |
|
doc = [ |
|
|
|
"black>=24.1.1", |
|
"mkdocs-material>=9.5.6", |
|
"mkdocstrings[python]>=0.24.0", |
|
"mkdocs-literate-nav>=0.6.1", |
|
] |
|
solutions = [ |
|
"huggingface-hub>=0.24.6", |
|
] |
|
|
|
[build-system] |
|
requires = ["hatchling", "hatch-vcs"] |
|
build-backend = "hatchling.build" |
|
|
|
[tool.hatch.version] |
|
source = "vcs" |
|
fallback-version = '0.0.0' |
|
|
|
[tool.rye] |
|
managed = true |
|
dev-dependencies = [ |
|
"pyright==1.1.372", |
|
"docformatter>=1.7.5", |
|
"pytest>=8.0.0", |
|
"coverage>=7.4.1", |
|
"typos>=1.18.2", |
|
"comfy-cli>=1.1.6", |
|
] |
|
|
|
|
|
[tool.hatch.metadata] |
|
allow-direct-references = true |
|
|
|
[tool.rye.scripts] |
|
serve-docs = "mkdocs serve" |
|
test-cov = "coverage run -m pytest" |
|
|
|
|
|
build-html-cov = { cmd = "coverage html", env = { PYTHONWARNINGS = "ignore:Couldn't parse::coverage.report_core" } } |
|
serve-cov-report = { chain = [ |
|
"build-html-cov", |
|
"python -m http.server 8080 -b 127.0.0.1 -d htmlcov", |
|
] } |
|
|
|
[tool.black] |
|
line-length = 120 |
|
|
|
[tool.ruff] |
|
src = ["src"] |
|
line-length = 120 |
|
|
|
[tool.ruff.lint] |
|
select = [ |
|
"I", |
|
] |
|
ignore = [ |
|
"F722", |
|
"E731", |
|
] |
|
|
|
[tool.ruff.lint.isort] |
|
|
|
|
|
|
|
|
|
combine-as-imports = true |
|
|
|
[tool.docformatter] |
|
black = true |
|
|
|
[tool.pyright] |
|
include = ["src/refiners", "tests", "scripts"] |
|
strict = ["*"] |
|
exclude = ["**/__pycache__", "tests/weights"] |
|
reportMissingTypeStubs = "warning" |
|
|
|
[tool.coverage.run] |
|
branch = true |
|
source = ["src/refiners"] |
|
|
|
|
|
[tool.coverage.report] |
|
ignore_errors = true |
|
exclude_also = [ |
|
"def __repr__", |
|
"raise NotImplementedError", |
|
"if TYPE_CHECKING:", |
|
"class .*\\bProtocol\\):", |
|
"@(abc\\.)?abstractmethod", |
|
] |
|
|
|
[tool.typos.default] |
|
extend-ignore-identifiers-re = ["NDArray*", "interm", "af000ded"] |
|
|
|
[tool.typos.default.extend-words] |
|
adaptee = "adaptee" |
|
|
|
[tool.pytest.ini_options] |
|
filterwarnings = [ |
|
"ignore::UserWarning:segment_anything_hq.modeling.tiny_vit_sam.*", |
|
"ignore::DeprecationWarning:timm.models.layers.*", |
|
"ignore::DeprecationWarning:timm.models.registry.*" |
|
] |
|
addopts = "--import-mode=importlib" |
|
|