suggested diffusers inference example
#10
by
linoyts
HF staff
- opened
Awesome work thanks for sharing it!
we added a community pipeline to diffusers that supports CFG to easily inference flux with CFG like this:
- download pipeline script (or alternatively change to your pipeline implementation :))
!wget https://raw.githubusercontent.com/huggingface/diffusers/main/examples/community/pipeline_flux_with_cfg.py
- load & prompt
import torch
import diffusers
from pipeline_flux_with_cfg import FluxCFGPipeline
pipe = FluxCFGPipeline.from_pretrained("ostris/OpenFLUX.1", torch_dtype=torch.bfloat16)
pipe(prompt="a tiny astronaut hatching from an egg").images[0]