Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-21 13:48:46 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

update getting started

Browse source
This commit is contained in:
Pierre Chapuis 2024-02-01 15:26:08 +01:00
parent a4c6975ac8
commit 5f474d01f8
2 changed files with 70 additions and 24 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
CONTRIBUTING.md
View file

@ -48,7 +48,7 @@ rye sync --all-features
Then, download and convert all the necessary weights. Be aware that this will use around 50 GB of disk space:
```bash
rye run python scripts/prepare_test_weights.py
python scripts/prepare_test_weights.py
```
Finally, run the tests:

92
docs/getting_started.md
View file

@ -2,42 +2,88 @@
Refiners is a micro framework on top of PyTorch with first class citizen APIs for foundation model adaptation.
## Installation
Refiners requires Python 3.10 or later, its main dependency is PyTorch.
### with git
## Recommended usage (development branch, with Rye)
To get the latest version of the code, clone the repository:
```bash
git clone https://github.com/finegrain-ai/refiners.git
```
Then install the package using pip:
Refiners is still a young project and development is active, so to use the latest and greatest version of the framework we recommend you use the `main` branch from our development repository.
Moreover, we recommend using [Rye](https://rye-up.com) which simplifies several things related to Python package management, so start by following the instructions to install it on your system.
### Trying Refiners, converting weights
To try Refiners, clone the GitHub repository and install it with all optional features:
```bash
git clone "git@github.com:finegrain-ai/refiners.git"
cd refiners
pip install .
rye sync --all-features
```
### with pip
Refiners is available on PyPI and can be installed using pip:
The format of state dicts used by Refiners is custom and we do not redistribute model weights, but we provide conversion tools and working scripts for popular models. For instance, let us convert the autoencoder from Stable Diffusion 1.5:
```bash
pip install refiners
python "scripts/conversion/convert_diffusers_autoencoder_kl.py" --to "lda.safetensors"
```
## Run foundational models and adapters
If you need to convert weights for all models, check out `script/prepare_test_weights.py` (warning: it requires a GPU with significant VRAM and a lot of disk space).
If you want to understand how to use Refiners with existing foundational models, please refer to the specific [Models](models/index.md) page.
Now let to check that it works copy your favorite 512x512 picture in the current directory as `input.png` and create `ldatest.py` with this content:
- [Stable Diffusion](/models/stable_diffusion)
- [Segment Anything](/models/segment_anything)
```py
from PIL import Image
from refiners.fluxion.utils import no_grad
from refiners.foundationals.latent_diffusion.stable_diffusion_1.model import SD1Autoencoder
## Write new foundational models and adapters
with no_grad():
lda = SD1Autoencoder()
lda.load_from_safetensors("lda.safetensors")
To understand how to write new adapters or models with Refiners, please have a look at the [Fluxion](fluxion/index.md) documentation.
image = Image.open("input.png")
latents = lda.encode_image(image)
decoded = lda.decode_latents(latents)
decoded.save("output.png")
```
Run it:
```bash
python ldatest.py
```
Inspect `output.png`: it should be similar to `input.png` but have a few differences. Latent Autoencoders are good compressors!
### Using refiners in your own project
So far you used Refiners as a standalone package, but if you want to create your own project using it as a dependency here is how you can proceed:
```bash
rye init --py "3.11" myproject
cd myproject
rye add --git "git@github.com:finegrain-ai/refiners.git" --features training refiners
rye sync
```
If you only intend to do inference and no training, you can drop `--features training`.
To convert weights, you can either use a copy of the `refiners` repository as described above or add the `conversion` feature as a development dependency:
```bash
rye add --dev --git "git@github.com:finegrain-ai/refiners.git" --features conversion refiners
```
Note that you will still need to download the conversion scripts independently if you go that route.
### What next?
We suggest you check out the [guides](/guides/) section to dive into the usage of Refiners, of the [Key Concepts](/concepts/chain/) section for a better understanding of how the framework works.
## Advanced usage
### Using other package managers (pip, Poetry...)
We use Rye to maintain and release Refiners but it conforms to the standard Python packaging guidelines and can be used with other package managers. Please refer to their respective documentation to figure out how to install a package from Git if you intend to use the development branch, as well as how to specify features.
### Using stable releases from PyPI
Although we recommend using our development branch, we do [publish more stable releases to PyPI](https://pypi.org/project/refiners/) and you are welcome to use them in your project. However, note that the format of weights can be different from the current state of the development branch, so you will need the conversion scripts from the corresponding tag in GitHub, for instance [here for v0.2.0](https://github.com/finegrain-ai/refiners/tree/v0.2.0).