metadata
license: mit
library_name: diffusers
pipeline_tag: text-to-image
tags:
- diffusers
- pmf
- image-generation
- class-conditional
- imagenet
inference: true
pMF-B-32-SIM
Self-contained Diffusers variant for pMF-B/32 (FD-SIM post-trained) (Pixel Mean Flows).
Recommended settings: guidance_scale=6.5, interval [0.1, 0.7], noise_scale=2.0.
Load
from pathlib import Path
from diffusers import DiffusionPipeline
import torch
model_dir = Path("./pMF-B-32-SIM")
pipe = DiffusionPipeline.from_pretrained(
str(model_dir),
local_files_only=True,
custom_pipeline=str(model_dir / "pipeline.py"),
trust_remote_code=True,
torch_dtype=torch.bfloat16,
).to("cuda")
image = pipe(
class_labels=207,
num_inference_steps=1,
guidance_scale=6.5,
guidance_interval_min=0.1,
guidance_interval_max=0.7,
noise_scale=2.0,
).images[0]