File size: 919 Bytes
ec0a9aa | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 | from __future__ import annotations
def add_fastercache_args(parser) -> None:
parser.add_argument(
"--use_fastercache",
action="store_true",
help="Enable FasterCache v1 acceleration.",
)
parser.add_argument(
"--fastercache_start_step",
type=int,
default=0,
help="First denoising step where FasterCache becomes eligible. Use 0 for ceil(0.3 * num_steps).",
)
parser.add_argument(
"--fastercache_model_interval",
type=int,
default=5,
help="Anchor interval for full model CFG passes.",
)
parser.add_argument(
"--fastercache_block_interval",
type=int,
default=3,
help="Anchor interval for block-level self-attention refresh.",
)
parser.add_argument(
"--fastercache_debug",
action="store_true",
help="Enable debug logs for FasterCache.",
)
|