|
--- |
|
license: other |
|
license_name: flux-1-dev-non-commercial-license |
|
license_link: https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICENSE.md |
|
language: |
|
- en |
|
base_model: |
|
- black-forest-labs/FLUX.1-dev |
|
library_name: diffusers |
|
--- |
|
|
|
## Model Details |
|
This is a diffusers version of [nyanko7/flux-dev-de-distill](https://huggingface.co/nyanko7/flux-dev-de-distill). |
|
|
|
## Usage |
|
```python |
|
import torch |
|
from diffusers import FluxTransformer2DModel |
|
from pipeline_flux_de_distill import FluxPipeline |
|
|
|
model_path = "black-forest-labs/FLUX.1-dev" |
|
|
|
transformer = FluxTransformer2DModel.from_pretrained( |
|
"InstantX/flux-dev-de-distill-diffusers", |
|
torch_dtype=torch.bfloat16 |
|
) |
|
|
|
pipeline = FluxPipeline.from_pretrained(model_path, transformer=transformer, torch_dtype=torch.bfloat16).to("cuda") |
|
|
|
prompt = "a tiny astronaut hatching from an egg on the moon" |
|
negative_prompt = "bad photo" |
|
|
|
image = pipeline( |
|
prompt=prompt, |
|
negative_prompt=negative_prompt, |
|
guidance_scale=3.5, |
|
num_inference_steps=24, |
|
).images[0] |
|
|
|
image.save("de-distill.jpg") |
|
``` |
|
|
|
## Other Resources |
|
- [ashen0209/Flux-Dev2Pro](https://huggingface.co/ashen0209/Flux-Dev2Pro) |
|
|
|
|
