FLUX.2
Collection
Our second generation of FLUX • 21 items • Updated • 212
How to use black-forest-labs/FLUX.2-dev-NVFP4 with Diffusers:
pip install -U diffusers transformers accelerate
import torch
from diffusers import DiffusionPipeline
from diffusers.utils import load_image
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.2-dev-NVFP4", dtype=torch.bfloat16, device_map="cuda")
prompt = "Turn this cat into a dog"
input_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png")
image = pipe(image=input_image, prompt=prompt).images[0]How to use black-forest-labs/FLUX.2-dev-NVFP4 with Diffusion Single File:
# No code snippets available yet for this library. # To use this model, check the repository files and the library's documentation. # Want to help? PRs adding snippets are welcome at: # https://github.com/huggingface/huggingface.js
FLUX.2 [dev] is a 32 billion parameter rectified flow transformer capable of generating, editing and combining images based on text instructions.
For more information, please read our blog post.
This repository holds an NVFP4 version and a mixed NVFP4/BF16 version of FLUX.2 [dev]. The mixed version has a small set of weights in BF16. The main repository of this model (full BF16 weights) can be found here.