Upload 8 files

.gitignore

@@ -0,0 +1,41 @@
+# compilation and distribution
+__pycache__/
+*.py[cod]
+dist/
+
+# virtual environments
+venv/
+
+# unit tests
+.pytest_cache/
+
+# tests' model weights
+tests/weights/
+tests/repos/
+tests/datasets/
+
+# ruff
+.ruff_cache
+
+# vscode
+.vscode
+
+# Weights & Biases (offline trainings)
+wandb/
+
+# macos
+.DS_Store
+
+# model weights
+*.safetensors
+
+# env variable definitions file
+.env
+
+# lock files
+requirements-dev.lock
+
+# coverage.py
+htmlcov/
+.coverage
+.coverage.*

CONTRIBUTING.md

@@ -0,0 +1,89 @@
+We are happy to accept contributions from everyone. Feel free to browse our [bounty list](https://www.finegrain.ai/bounties) to find a task you would like to work on.
+
+This document describes the process for contributing to Refiners.
+
+## Licensing
+
+Refiners is a library that is freely available to use and modify under the MIT License. It's essential to exercise caution when using external code, as any code that can affect the licensing of Refiners, including proprietary code, should not be copied and pasted. It's worth noting that some open-source licenses can also be problematic. For instance, we'd need to redistribute an Apache 2.0 license if you're using code from an Apache 2.0 codebase.
+
+## Design principles
+
+We do not enforce strict rules on the design of the code, but we do have a few guidelines that we try to follow:
+
+- No dead code. We keep the codebase clean and remove unnecessary code/functionality.
+- No unnecessary dependencies. We keep the number of dependencies to a minimum and only add new ones if necessary.
+- Separate concerns. We separate the code into different modules and avoid having too many dependencies between modules. In particular, we try not to revisit existing code/models when adding new functionality. Instead, we add new functionality in a separate module with the `Adapter` pattern.
+- Declarative style. We make the code as declarative, self-documenting, and easily read as possible. By reading the model's `repr`, you should understand how it works. We use explicit names for the different components of the models or the variables in the code.
+
+## Setting up your environment
+
+We use [Rye](https://rye-up.com/guide/installation/) to manage our development environment. Please follow the instructions on the Rye website to install it.
+
+Once Rye is installed, you can clone the repository and run `rye sync` to install the dependencies.
+
+## Linting
+
+We use the standard integration of [ruff](https://docs.astral.sh/ruff/) in Rye to lint and format our code. You can lint your code by running:
+
+```bash
+rye fmt
+rye lint --fix
+```
+
+We also enforce strict type checking with [pyright](https://github.com/microsoft/pyright). You can run the type checker with:
+
+```bash
+rye run pyright
+```
+
+## Running the tests
+
+Running end-to-end tests is pretty compute-intensive, and you must convert all the model weights to the correct format before you can run them.
+
+First, install test dependencies with:
+
+```bash
+rye sync --all-features
+```
+
+Then, download and convert all the necessary weights. Be aware that this will use around 100 GB of disk space:
+
+```bash
+python scripts/prepare_test_weights.py
+```
+
+Some tests require cloning the original implementation of the model as they use `torch.hub.load`:
+
+```bash
+git clone [email protected]:facebookresearch/dinov2.git tests/repos/dinov2
+```
+
+Finally, run the tests:
+
+```bash
+rye run pytest
+```
+
+The `-k` option is handy to run a subset of tests that match a given expression, e.g.:
+
+```bash
+rye run pytest -k diffusion_std_init_image
+```
+
+You can enforce running tests on CPU. Tests that require a GPU will be skipped.
+
+```bash
+REFINERS_TEST_DEVICE=cpu rye run pytest
+```
+
+You can collect [code coverage](https://github.com/nedbat/coveragepy) data while running tests with, e.g.:
+
+```bash
+rye run test-cov
+```
+
+Then, browse the corresponding HTML report with:
+
+```bash
+rye run serve-cov-report
+```

LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2023 to present Lagon Technologies and individual contributors.
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

README.md

@@ -1,12 +1,107 @@
----
-title: Image Enhancer
-emoji: 🚀
-colorFrom: gray
-colorTo: yellow
-sdk: gradio
-sdk_version: 4.44.0
-app_file: app.py
-pinned: false
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+<div align="center">
+
+<picture>
+  <source media="(prefers-color-scheme: dark)" srcset="https://raw.githubusercontent.com/finegrain-ai/refiners/main/assets/logo_dark.png">
+  <source media="(prefers-color-scheme: light)" srcset="https://raw.githubusercontent.com/finegrain-ai/refiners/main/assets/logo_light.png">
+  <img alt="Finegrain Refiners Library" width="352" height="128" style="max-width: 100%;">
+</picture>
+
+**The simplest way to train and run adapters on top of foundation models**
+
+[**Manifesto**](https://refine.rs/home/why/) |
+[**Docs**](https://refine.rs) |
+[**Guides**](https://refine.rs/guides/adapting_sdxl/) |
+[**Discussions**](https://github.com/finegrain-ai/refiners/discussions) |
+[**Discord**](https://discord.gg/mCmjNUVV7d)
+
+______________________________________________________________________
+
+[![dependencies - Rye](https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/rye/main/artwork/badge.json)](https://github.com/astral-sh/rye)
+[![linting - Ruff](https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/ruff/main/assets/badge/v2.json)](https://github.com/astral-sh/ruff)
+[![packaging - Hatch](https://img.shields.io/badge/%F0%9F%A5%9A-Hatch-4051b5.svg)](https://github.com/pypa/hatch)
+[![PyPI - Python Version](https://img.shields.io/pypi/pyversions/refiners)](https://pypi.org/project/refiners/)
+[![PyPI - Status](https://badge.fury.io/py/refiners.svg)](https://pypi.org/project/refiners/)
+[![license](https://img.shields.io/badge/license-MIT-blue)](/LICENSE) \
+[![code bounties](https://img.shields.io/badge/code-bounties-blue)](https://finegrain.ai/bounties)
+[![Discord](https://img.shields.io/discord/1179456777406922913?logo=discord&logoColor=white&color=%235765F2)](https://discord.gg/mCmjNUVV7d)
+[![HuggingFace - Refiners](https://img.shields.io/badge/refiners-ffd21e?logo=huggingface&labelColor=555)](https://huggingface.co/refiners)
+[![HuggingFace - Finegrain](https://img.shields.io/badge/finegrain-ffd21e?logo=huggingface&labelColor=555)](https://huggingface.co/finegrain)
+[![ComfyUI Registry](https://img.shields.io/badge/ComfyUI_Registry-comfyui--refiners-1a56db)](https://registry.comfy.org/publishers/finegrain/nodes/comfyui-refiners)
+
+</div>
+
+## Latest News 🔥
+
+- Added [ELLA](https://arxiv.org/abs/2403.05135) for better prompts handling (contributed by [@ily-R](https://github.com/ily-R))
+- Added the Box Segmenter all-in-one solution ([model](https://huggingface.co/finegrain/finegrain-box-segmenter), [HF Space](https://huggingface.co/spaces/finegrain/finegrain-object-cutter))
+- Added [MVANet](https://arxiv.org/abs/2404.07445) for high resolution segmentation
+- Added [IC-Light](https://github.com/lllyasviel/IC-Light) to manipulate the illumination of images
+- Added Multi Upscaler for high-resolution image generation, inspired from [Clarity Upscaler](https://github.com/philz1337x/clarity-upscaler) ([HF Space](https://huggingface.co/spaces/finegrain/enhancer))
+- Added [HQ-SAM](https://arxiv.org/abs/2306.01567) for high quality mask prediction with Segment Anything
+- Added [SDXL-Lightning](https://arxiv.org/abs/2402.13929)
+- Added [Latent Consistency Models](https://arxiv.org/abs/2310.04378) and [LCM-LoRA](https://arxiv.org/abs/2311.05556) for Stable Diffusion XL
+- Added [Style Aligned adapter](https://arxiv.org/abs/2312.02133) to Stable Diffusion models
+- Added [ControlLoRA (v2) adapter](https://github.com/HighCWu/control-lora-v2) to Stable Diffusion XL
+- Added [Euler's method](https://arxiv.org/abs/2206.00364) to solvers (contributed by [@israfelsr](https://github.com/israfelsr))
+- Added [DINOv2](https://github.com/facebookresearch/dinov2) for high-performance visual features (contributed by [@Laurent2916](https://github.com/Laurent2916))
+- Added [FreeU](https://github.com/ChenyangSi/FreeU) for improved quality at no cost (contributed by [@isamu-isozaki](https://github.com/isamu-isozaki))
+- Added [Restart Sampling](https://github.com/Newbeeer/diffusion_restart_sampling) for improved image generation ([example](https://github.com/Newbeeer/diffusion_restart_sampling/issues/4))
+- Added [Self-Attention Guidance](https://github.com/KU-CVLAB/Self-Attention-Guidance/) to avoid e.g. too smooth images ([example](https://github.com/SusungHong/Self-Attention-Guidance/issues/4))
+- Added [T2I-Adapter](https://github.com/TencentARC/T2I-Adapter) for extra guidance ([example](https://github.com/TencentARC/T2I-Adapter/discussions/93))
+- Added [MultiDiffusion](https://github.com/omerbt/MultiDiffusion) for e.g. panorama images
+- Added [IP-Adapter](https://github.com/tencent-ailab/IP-Adapter), aka image prompt ([example](https://github.com/tencent-ailab/IP-Adapter/issues/92))
+- Added [Segment Anything](https://github.com/facebookresearch/segment-anything) to foundation models
+- Added [SDXL 1.0](https://github.com/Stability-AI/generative-models) to foundation models
+- Made possible to add new concepts to the CLIP text encoder, e.g. via [Textual Inversion](https://arxiv.org/abs/2208.01618)
+
+## Installation
+
+The current recommended way to install Refiners is from source using [Rye](https://rye-up.com/):
+
+```bash
+git clone "[email protected]:finegrain-ai/refiners.git"
+cd refiners
+rye sync --all-features
+```
+
+## Documentation
+
+Refiners comes with a MkDocs-based documentation website available at https://refine.rs. You will find there a [quick start guide](https://refine.rs/getting-started/recommended/), a description of the [key concepts](https://refine.rs/concepts/chain/), as well as in-depth foundation model adaptation [guides](https://refine.rs/guides/adapting_sdxl/).
+
+## Awesome Adaptation Papers
+
+If you're interested in understanding the diversity of use cases for foundation model adaptation (potentially beyond the specific adapters supported by Refiners), we suggest you take a look at these outstanding papers:
+
+- [ControlNet](https://arxiv.org/abs/2302.05543)
+- [T2I-Adapter](https://arxiv.org/abs/2302.08453)
+- [IP-Adapter](https://arxiv.org/abs/2308.06721)
+- [Medical SAM Adapter](https://arxiv.org/abs/2304.12620)
+- [3DSAM-adapter](https://arxiv.org/abs/2306.13465)
+- [SAM-adapter](https://arxiv.org/abs/2304.09148)
+- [Cross Modality Attention Adapter](https://arxiv.org/abs/2307.01124)
+- [UniAdapter](https://arxiv.org/abs/2302.06605)
+
+## Projects using Refiners
+
+- https://github.com/brycedrennan/imaginAIry
+
+## Credits
+
+We took inspiration from these great projects:
+
+- [tinygrad](https://github.com/tinygrad/tinygrad) - For something between PyTorch and [karpathy/micrograd](https://github.com/karpathy/micrograd)
+- [Composer](https://github.com/mosaicml/composer) - A PyTorch Library for Efficient Neural Network Training
+- [Keras](https://github.com/keras-team/keras) - Deep Learning for humans
+
+## Citation
+
+```bibtex
+@misc{the-finegrain-team-2023-refiners,
+  author = {Benjamin Trom and Pierre Chapuis and Cédric Deltheil},
+  title = {Refiners: The simplest way to train and run adapters on top of foundation models},
+  year = {2023},
+  publisher = {GitHub},
+  journal = {GitHub repository},
+  howpublished = {\url{https://github.com/finegrain-ai/refiners}}
+}
+```

mkdocs.yml

@@ -0,0 +1,104 @@
+site_name: Refiners
+site_description: A micro framework on top of PyTorch with first class citizen APIs for foundation model adaptation
+repo_name: Refiners
+repo_url: https://github.com/finegrain-ai/refiners
+edit_uri: edit/main/docs/
+copyright: © Lagon Technologies
+theme:
+  favicon: assets/favicon.svg
+  logo: assets/favicon.svg
+  name: material
+  custom_dir: docs/overrides
+  palette:
+    primary: deep orange
+    accent: deep orange
+  features:
+    - navigation.tabs
+    - navigation.sections
+    - navigation.top
+    - navigation.tracking
+    - navigation.expand
+    - navigation.path
+    - toc.follow
+    - navigation.tabs.sticky
+    - content.code.copy
+    - announce.dismiss
+plugins:
+  - search
+  - tags
+  - literate-nav:
+      nav_file: SUMMARY.md
+  - mkdocstrings:
+      handlers:
+        python:
+          import:
+            - https://docs.python.org/3/objects.inv
+            - https://pytorch.org/docs/master/objects.inv
+            - https://docs.kidger.site/jaxtyping/objects.inv
+          options:
+            show_bases: true
+            show_source: true
+            show_root_toc_entry: false
+            show_symbol_type_heading: true
+            show_symbol_type_toc: true
+            show_if_no_docstring: false
+            filters:
+              - "!^_"
+            docstring_options:
+              ignore_init_summary: true
+            merge_init_into_class: true
+            separate_signature: true
+            show_signature_annotations: true
+            signature_crossrefs: true
+watch:
+  - src/refiners
+extra_css:
+  - stylesheets/extra.css
+nav:
+  - Home: 
+    - Welcome: index.md
+    - Manifesto: home/why.md
+  - Getting started:
+    - getting-started/recommended.md
+    - getting-started/advanced.md
+  - Key Concepts:
+    - concepts/chain.md
+    - concepts/context.md
+    - concepts/adapter/index.md
+  - Guides:
+    - Adapting SDXL: guides/adapting_sdxl/index.md
+    - Training 101: guides/training_101/index.md
+  - API Reference:
+      - Refiners: reference/
+extra:
+  social:
+    - icon: fontawesome/brands/discord
+      link: https://discord.gg/mCmjNUVV7d
+    - icon: fontawesome/brands/github
+      link: https://github.com/finegrain-ai/refiners   
+    - icon: fontawesome/brands/twitter
+      link: https://twitter.com/finegrain_ai 
+    - icon: fontawesome/brands/linkedin
+      link: https://www.linkedin.com/company/finegrain-ai/
+markdown_extensions:
+  - admonition
+  - attr_list
+  - footnotes
+  - pymdownx.highlight:
+      anchor_linenums: true
+      line_spans: __span
+      pygments_lang_class: true
+  - pymdownx.inlinehilite
+  - pymdownx.snippets
+  - pymdownx.superfences
+  - pymdownx.details
+  - pymdownx.tabbed:
+      alternate_style: true
+  - pymdownx.emoji:
+      emoji_index: !!python/name:material.extensions.emoji.twemoji
+      emoji_generator: !!python/name:material.extensions.emoji.to_svg
+  - toc:
+      permalink: true
+validation:
+  links:
+    absolute_links: info  # use relative_to_docs when we switch to mkdocs 1.6

pyproject.toml

@@ -0,0 +1,172 @@
+[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",
+    # An unofficial Python package for Meta AI's Segment Anything Model:
+    # https://github.com/opengeos/segment-anything
+    "segment-anything-py>=1.0",
+    # Official Python package for HQ-SAM
+    "segment-anything-hq>=0.3",
+    # HQ-SAM missing dependency:
+    # https://github.com/SysCV/sam-hq/pull/59
+    "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 = [
+    # required by mkdocs to format the signatures
+    "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"
+# Work around for "Couldn't parse" errors due to e.g. opencv-python:
+# https://github.com/nedbat/coveragepy/issues/1653
+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"] # see https://docs.astral.sh/ruff/settings/#src
+line-length = 120
+
+[tool.ruff.lint]
+select = [
+    "I", # isort
+]
+ignore = [
+    "F722", # forward-annotation-syntax-error, because of Jaxtyping
+    "E731", # do-not-assign-lambda
+]
+
+[tool.ruff.lint.isort]
+# Allow this kind of import on a single line:
+#
+#     from torch import device as Device, dtype as DType
+#
+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"]
+
+# Also apply to HTML output, where appropriate
+[tool.coverage.report]
+ignore_errors = true # see `build-html-cov` for details
+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"  # Common name for an adapter's target
+
+[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"

requirements.docs.txt

@@ -0,0 +1,2 @@
+-e file:.[training]
+-e file:.[doc]

requirements.lock

@@ -0,0 +1,442 @@
+# generated by rye
+# use `rye lock` or `rye sync` to update this lockfile
+#
+# last locked with the following flags:
+#   pre: false
+#   features: []
+#   all-features: true
+#   with-sources: false
+#   generate-hashes: false
+#   universal: false
+
+-e file:.
+aiohappyeyeballs==2.3.4
+    # via aiohttp
+aiohttp==3.10.0
+    # via datasets
+    # via fsspec
+aiosignal==1.3.1
+    # via aiohttp
+annotated-types==0.7.0
+    # via pydantic
+arrow==1.3.0
+    # via isoduration
+async-timeout==4.0.3
+    # via aiohttp
+attrs==23.2.0
+    # via aiohttp
+    # via jsonschema
+    # via referencing
+babel==2.15.0
+    # via mkdocs-material
+backports-strenum==1.3.1
+    # via griffe
+beautifulsoup4==4.12.3
+    # via gdown
+bitsandbytes==0.43.3
+    # via refiners
+black==24.4.2
+    # via refiners
+boto3==1.34.152
+    # via neptune
+botocore==1.34.152
+    # via boto3
+    # via s3transfer
+bravado==11.0.3
+    # via neptune
+bravado-core==6.1.1
+    # via bravado
+certifi==2024.7.4
+    # via requests
+    # via sentry-sdk
+charset-normalizer==3.3.2
+    # via requests
+click==8.1.7
+    # via black
+    # via mkdocs
+    # via mkdocstrings
+    # via neptune
+    # via wandb
+colorama==0.4.6
+    # via griffe
+    # via mkdocs-material
+datasets==2.20.0
+    # via refiners
+diffusers==0.29.2
+    # via refiners
+dill==0.3.8
+    # via datasets
+    # via multiprocess
+docker-pycreds==0.4.0
+    # via wandb
+filelock==3.15.4
+    # via datasets
+    # via diffusers
+    # via gdown
+    # via huggingface-hub
+    # via torch
+    # via transformers
+    # via triton
+fqdn==1.5.1
+    # via jsonschema
+frozenlist==1.4.1
+    # via aiohttp
+    # via aiosignal
+fsspec==2024.5.0
+    # via datasets
+    # via huggingface-hub
+    # via torch
+future==1.0.0
+    # via neptune
+gdown==5.2.0
+    # via refiners
+ghp-import==2.1.0
+    # via mkdocs
+gitdb==4.0.11
+    # via gitpython
+gitpython==3.1.43
+    # via neptune
+    # via refiners
+    # via wandb
+griffe==0.48.0
+    # via mkdocstrings-python
+huggingface-hub==0.24.6
+    # via datasets
+    # via diffusers
+    # via refiners
+    # via timm
+    # via tokenizers
+    # via transformers
+idna==3.7
+    # via jsonschema
+    # via requests
+    # via yarl
+importlib-metadata==8.2.0
+    # via diffusers
+importlib-resources==6.4.0
+    # via swagger-spec-validator
+isoduration==20.11.0
+    # via jsonschema
+jaxtyping==0.2.33
+    # via refiners
+jinja2==3.1.4
+    # via mkdocs
+    # via mkdocs-material
+    # via mkdocstrings
+    # via torch
+jmespath==1.0.1
+    # via boto3
+    # via botocore
+jsonpointer==3.0.0
+    # via jsonschema
+jsonref==1.1.0
+    # via bravado-core
+jsonschema==4.23.0
+    # via bravado-core
+    # via swagger-spec-validator
+jsonschema-specifications==2023.12.1
+    # via jsonschema
+loguru==0.7.2
+    # via refiners
+markdown==3.6
+    # via mkdocs
+    # via mkdocs-autorefs
+    # via mkdocs-material
+    # via mkdocstrings
+    # via pymdown-extensions
+markupsafe==2.1.5
+    # via jinja2
+    # via mkdocs
+    # via mkdocs-autorefs
+    # via mkdocstrings
+mergedeep==1.3.4
+    # via mkdocs
+    # via mkdocs-get-deps
+mkdocs==1.6.0
+    # via mkdocs-autorefs
+    # via mkdocs-literate-nav
+    # via mkdocs-material
+    # via mkdocstrings
+mkdocs-autorefs==1.0.1
+    # via mkdocstrings
+mkdocs-get-deps==0.2.0
+    # via mkdocs
+mkdocs-literate-nav==0.6.1
+    # via refiners
+mkdocs-material==9.5.30
+    # via refiners
+mkdocs-material-extensions==1.3.1
+    # via mkdocs-material
+mkdocstrings==0.25.2
+    # via mkdocstrings-python
+    # via refiners
+mkdocstrings-python==1.10.7
+    # via mkdocstrings
+monotonic==1.6
+    # via bravado
+mpmath==1.3.0
+    # via sympy
+msgpack==1.0.8
+    # via bravado
+    # via bravado-core
+multidict==6.0.5
+    # via aiohttp
+    # via yarl
+multiprocess==0.70.16
+    # via datasets
+mypy-extensions==1.0.0
+    # via black
+neptune==1.10.4
+    # via refiners
+networkx==3.3
+    # via torch
+numpy==2.0.1
+    # via bitsandbytes
+    # via datasets
+    # via diffusers
+    # via pandas
+    # via pyarrow
+    # via refiners
+    # via torchvision
+    # via transformers
+nvidia-cublas-cu12==12.1.3.1
+    # via nvidia-cudnn-cu12
+    # via nvidia-cusolver-cu12
+    # via torch
+nvidia-cuda-cupti-cu12==12.1.105
+    # via torch
+nvidia-cuda-nvrtc-cu12==12.1.105
+    # via torch
+nvidia-cuda-runtime-cu12==12.1.105
+    # via torch
+nvidia-cudnn-cu12==9.1.0.70
+    # via torch
+nvidia-cufft-cu12==11.0.2.54
+    # via torch
+nvidia-curand-cu12==10.3.2.106
+    # via torch
+nvidia-cusolver-cu12==11.4.5.107
+    # via torch
+nvidia-cusparse-cu12==12.1.0.106
+    # via nvidia-cusolver-cu12
+    # via torch
+nvidia-nccl-cu12==2.20.5
+    # via torch
+nvidia-nvjitlink-cu12==12.6.20
+    # via nvidia-cusolver-cu12
+    # via nvidia-cusparse-cu12
+nvidia-nvtx-cu12==12.1.105
+    # via torch
+oauthlib==3.2.2
+    # via neptune
+    # via requests-oauthlib
+packaging==24.1
+    # via black
+    # via datasets
+    # via huggingface-hub
+    # via mkdocs
+    # via neptune
+    # via refiners
+    # via transformers
+paginate==0.5.6
+    # via mkdocs-material
+pandas==2.2.2
+    # via datasets
+    # via neptune
+pathspec==0.12.1
+    # via black
+    # via mkdocs
+pillow==10.4.0
+    # via diffusers
+    # via neptune
+    # via refiners
+    # via torchvision
+piq==0.8.0
+    # via refiners
+platformdirs==4.2.2
+    # via black
+    # via mkdocs-get-deps
+    # via mkdocstrings
+    # via wandb
+prodigyopt==1.0
+    # via refiners
+protobuf==5.27.3
+    # via wandb
+psutil==6.0.0
+    # via neptune
+    # via wandb
+pyarrow==17.0.0
+    # via datasets
+pyarrow-hotfix==0.6
+    # via datasets
+pydantic==2.8.2
+    # via refiners
+pydantic-core==2.20.1
+    # via pydantic
+pygments==2.18.0
+    # via mkdocs-material
+pyjwt==2.9.0
+    # via neptune
+pymdown-extensions==10.9
+    # via mkdocs-material
+    # via mkdocstrings
+pysocks==1.7.1
+    # via requests
+python-dateutil==2.9.0.post0
+    # via arrow
+    # via botocore
+    # via bravado
+    # via bravado-core
+    # via ghp-import
+    # via pandas
+pytz==2024.1
+    # via bravado-core
+    # via pandas
+pyyaml==6.0.1
+    # via bravado
+    # via bravado-core
+    # via datasets
+    # via huggingface-hub
+    # via mkdocs
+    # via mkdocs-get-deps
+    # via pymdown-extensions
+    # via pyyaml-env-tag
+    # via swagger-spec-validator
+    # via timm
+    # via transformers
+    # via wandb
+pyyaml-env-tag==0.1
+    # via mkdocs
+referencing==0.35.1
+    # via jsonschema
+    # via jsonschema-specifications
+regex==2024.7.24
+    # via diffusers
+    # via mkdocs-material
+    # via transformers
+requests==2.32.3
+    # via bravado
+    # via bravado-core
+    # via datasets
+    # via diffusers
+    # via gdown
+    # via huggingface-hub
+    # via mkdocs-material
+    # via neptune
+    # via refiners
+    # via requests-oauthlib
+    # via transformers
+    # via wandb
+requests-oauthlib==2.0.0
+    # via neptune
+rfc3339-validator==0.1.4
+    # via jsonschema
+rfc3986-validator==0.1.1
+    # via jsonschema
+rpds-py==0.19.1
+    # via jsonschema
+    # via referencing
+s3transfer==0.10.2
+    # via boto3
+safetensors==0.4.5
+    # via diffusers
+    # via refiners
+    # via timm
+    # via transformers
+segment-anything-hq==0.3
+    # via refiners
+segment-anything-py==1.0.1
+    # via refiners
+sentencepiece==0.2.0
+    # via refiners
+sentry-sdk==2.12.0
+    # via wandb
+setproctitle==1.3.3
+    # via wandb
+setuptools==72.1.0
+    # via wandb
+simplejson==3.19.2
+    # via bravado
+    # via bravado-core
+six==1.16.0
+    # via bravado
+    # via bravado-core
+    # via docker-pycreds
+    # via neptune
+    # via python-dateutil
+    # via rfc3339-validator
+smmap==5.0.1
+    # via gitdb
+soupsieve==2.6
+    # via beautifulsoup4
+swagger-spec-validator==3.0.4
+    # via bravado-core
+    # via neptune
+sympy==1.13.1
+    # via torch
+timm==1.0.8
+    # via refiners
+tokenizers==0.19.1
+    # via transformers
+tomli==2.0.1
+    # via black
+    # via refiners
+torch==2.4.0
+    # via bitsandbytes
+    # via refiners
+    # via segment-anything-hq
+    # via segment-anything-py
+    # via timm
+    # via torchvision
+torchvision==0.19.0
+    # via piq
+    # via refiners
+    # via segment-anything-hq
+    # via segment-anything-py
+    # via timm
+tqdm==4.66.4
+    # via datasets
+    # via gdown
+    # via huggingface-hub
+    # via refiners
+    # via transformers
+transformers==4.43.3
+    # via refiners
+triton==3.0.0
+    # via torch
+typeguard==2.13.3
+    # via jaxtyping
+types-python-dateutil==2.9.0.20240316
+    # via arrow
+typing-extensions==4.12.2
+    # via black
+    # via bravado
+    # via huggingface-hub
+    # via neptune
+    # via pydantic
+    # via pydantic-core
+    # via swagger-spec-validator
+    # via torch
+tzdata==2024.1
+    # via pandas
+uri-template==1.3.0
+    # via jsonschema
+urllib3==2.2.2
+    # via botocore
+    # via neptune
+    # via requests
+    # via sentry-sdk
+wandb==0.17.5
+    # via refiners
+watchdog==4.0.1
+    # via mkdocs
+webcolors==24.6.0
+    # via jsonschema
+websocket-client==1.8.0
+    # via neptune
+xxhash==3.4.1
+    # via datasets
+yarl==1.9.4
+    # via aiohttp
+zipp==3.19.2
+    # via importlib-metadata