Views
No views yet
pipeline.py — PMFPipelinescheduler/scheduler_config.json — FlowMatchEulerDiscreteScheduler configtransformer/transformer_pmf.py — PMFTransformer2DModeltransformer/ — converted weights and config| Checkpoint | Path | Resolution | Recommended CFG (ω) | CFG interval | Noise scale |
|---|---|---|---|---|---|
| pMF-B/16 | ./pMF-B-16 | 256×256 | 7.5 | [0.1, 0.8] | 1.0 |
| pMF-B/32 | ./pMF-B-32 | 512×512 | 6.5 | [0.1, 0.7] | 2.0 |
| pMF-L/16 | ./pMF-L-16 | 256×256 | 7.0 | [0.2, 0.7] | 1.0 |
| pMF-L/32 | ./pMF-L-32 | 512×512 | 7.5 | [0.2, 0.6] | 4.0 |
| pMF-H/16 | ./pMF-H-16 | 256×256 | 7.0 | [0.2, 0.6] | 2.0 |
| pMF-H/32 | ./pMF-H-32 | 512×512 | 5.5 | [0.1, 0.6] | 4.0 |
1from pathlib import Path
2from diffusers import DiffusionPipeline
3import torch
4
5model_dir = Path("./pMF-L-16")
6pipe = DiffusionPipeline.from_pretrained(
7 str(model_dir),
8 local_files_only=True,
9 custom_pipeline=str(model_dir / "pipeline.py"),
10 trust_remote_code=True,
11 torch_dtype=torch.float32,
12).to("cuda")
13
14generator = torch.Generator(device="cuda").manual_seed(42)
15image = pipe(
16 class_labels="golden retriever",
17 num_inference_steps=1,
18 guidance_scale=7.0,
19 guidance_interval_min=0.2,
20 guidance_interval_max=0.7,
21 noise_scale=1.0,
22 generator=generator,
23).images[0]
24image.save("demo.png")./pMF-L-16), not the repo root.