---
license: mit
library_name: diffusers
pipeline_tag: text-to-image
tags:
- diffusers
- imf
- image-generation
- class-conditional
---

# iMF-B-SIM

Self-contained Diffusers variant for **iMF-B/2 (FD-SIM post-trained)** (Improved Mean Flows).

## Load

```python
from pathlib import Path
from diffusers import DiffusionPipeline
import torch

model_dir = Path("iMF-B-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.float32,
)
pipe.to("cuda")

image = pipe(
    class_labels=207,
    num_inference_steps=1,
    guidance_scale=8.0,
    guidance_interval_start=0.4,
    guidance_interval_end=0.65,
).images[0]
```