| license: mit | |
| library_name: diffusers | |
| pipeline_tag: text-to-image | |
| tags: | |
| - diffusers | |
| - imf | |
| - image-generation | |
| - class-conditional | |
| # iMF-B-2 | |
| Self-contained Diffusers variant for **iMF-B/2** (Improved Mean Flows). | |
| ## Load | |
| ```python | |
| from pathlib import Path | |
| from diffusers import DiffusionPipeline | |
| import torch | |
| model_dir = Path("iMF-B-2") | |
| 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=2.7, | |
| guidance_interval_start=0.1, | |
| guidance_interval_end=0.9, | |
| ).images[0] | |
| ``` | |