| | --- |
| | language: |
| | - en |
| | base_model: |
| | - black-forest-labs/FLUX.1-dev |
| | tags: |
| | - art |
| | --- |
| | # Flux.1Fill-dev-nf4 |
| |
|
| | This is a custom **Flux.1 Fill pipeline** that combines: |
| |
|
| | - Base pipeline: [`black-forest-labs/FLUX.1-dev`](https://huggingface.co/black-forest-labs/FLUX.1-dev) |
| | - Transformer: [`sayakpaul/FLUX.1-Fill-dev-nf4`](https://huggingface.co/sayakpaul/FLUX.1-Fill-dev-nf4) |
| | - Text Encoder 2: [`sayakpaul/FLUX.1-Fill-dev-nf4`](https://huggingface.co/sayakpaul/FLUX.1-Fill-dev-nf4) |
| |
|
| | ## Usage |
| |
|
| | ```python |
| | from diffusers import DiffusionPipeline |
| | import torch |
| | |
| | pipe = DiffusionPipeline.from_pretrained( |
| | "CutePickle/FLUX.1-Fill-dev-nf4", |
| | torch_dtype=torch.bfloat16 |
| | ) |
| | pipe.enable_model_cpu_offload() |
| | |
| | image = pipe( |
| | prompt="A cyberpunk cityscape at night", |
| | image=init_image, |
| | mask_image=mask |
| | ).images[0] |
| | |
| | image.save("output.png") |