| """HoLo-FuSe CLI: python generate.py --label Cat --steps 16 --cfg 1.6 --seed 0 --out cat.png""" | |
| import argparse | |
| from model import generate | |
| ap = argparse.ArgumentParser() | |
| ap.add_argument("--label", default="Cat", choices=["Cat", "Dog"]) | |
| ap.add_argument("--arm", default="hsl", choices=["hsl", "learned", "none"]) | |
| ap.add_argument("--steps", type=int, default=16) | |
| ap.add_argument("--cfg", type=float, default=1.6) | |
| ap.add_argument("--seed", type=int, default=0) | |
| ap.add_argument("--out", default="sample.png") | |
| a = ap.parse_args() | |
| img = generate(a.label, a.arm, a.steps, a.cfg, a.seed)[0] | |
| img.save(a.out) | |
| print(f"saved {a.out} ({a.label}, {a.arm}, steps={a.steps}, cfg={a.cfg}, seed={a.seed})") | |