| license: mit | |
| library_name: diffusers | |
| pipeline_tag: text-to-image | |
| tags: | |
| - diffusers | |
| - pmf | |
| - image-generation | |
| - class-conditional | |
| - imagenet | |
| inference: true | |
| # pMF-L-16 | |
| Self-contained Diffusers variant for **pMF-L/16** (Pixel Mean Flows). | |
| Recommended settings: `guidance_scale=7.0`, interval `[0.2, 0.7]`, `noise_scale=1.0`. | |
| ## Load | |
| ```python | |
| from pathlib import Path | |
| from diffusers import DiffusionPipeline | |
| import torch | |
| model_dir = Path("./pMF-L-16") | |
| 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.float32, | |
| ).to("cuda") | |
| image = pipe( | |
| class_labels=207, | |
| num_inference_steps=1, | |
| guidance_scale=7.0, | |
| guidance_interval_min=0.2, | |
| guidance_interval_max=0.7, | |
| noise_scale=1.0, | |
| ).images[0] | |
| ``` | |