FLUX.2
Collection
Our second generation of FLUX • 21 items • Updated • 215
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]
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.
- Downloads last month
- 11,031