metadata
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
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]