black-forest-labs

/

FLUX.1-dev

like 9.41k

Follow

Black Forest Labs 13.4k

import torch
from diffusers import FluxPipeline

model_id = "black-forest-labs/FLUX.1-dev" #you can also use black-forest-labs/FLUX.1-dev

pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16, cache_dir="D:/my_model_cache")
pipe.enable_model_cpu_offload() #save some VRAM by offloading the model to CPU. Remove this if you have enough GPU power

prompt = "A cat holding a sign that says hello world"
seed = 42
image = pipe(
prompt,
output_type="pil",
num_inference_steps=4, #use a larger number if you are using [dev]
generator=torch.Generator("cpu").manual_seed(seed)
).images[0]
image.save("flux-schnell.png")

Prompt after running：
Loading pipeline components...: 0%| | 0/7 [00:00<?, ?it/s]You set add_prefix_space. The tokenizer needs to be converted from the slow tokenizers
Loading pipeline components...: 29%|██████████████████▊ | 2/7 [00:00<00:00, 6.76it/s]

Nothing happened