File size: 863 Bytes
c8b6e26 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 | ---
license: mit
library_name: diffusers
pipeline_tag: text-to-image
tags:
- diffusers
- pmf
- image-generation
- class-conditional
- imagenet
inference: true
---
# pMF-B-16
Self-contained Diffusers variant for **pMF-B/16** (Pixel Mean Flows).
Recommended settings: `guidance_scale=7.5`, interval `[0.1, 0.8]`, `noise_scale=1.0`.
## Load
```python
from pathlib import Path
from diffusers import DiffusionPipeline
import torch
model_dir = Path("./pMF-B-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.5,
guidance_interval_min=0.1,
guidance_interval_max=0.8,
noise_scale=1.0,
).images[0]
```
|