File size: 776 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 | from __future__ import annotations
def add_dicache_args(parser) -> None:
parser.add_argument(
"--use_dicache",
action="store_true",
help="Enable DiCache v1 acceleration for Ctrl-World.",
)
parser.add_argument(
"--dicache_rel_l1_thresh",
type=float,
default=0.08,
help="Relative L1 threshold for DiCache skip decisions.",
)
parser.add_argument(
"--dicache_ret_ratio",
type=float,
default=0.2,
help="Fraction of denoising steps to keep dense before DiCache becomes eligible.",
)
parser.add_argument(
"--dicache_probe_depth",
type=int,
default=2,
help="Number of attention-bearing UNet stages used for the DiCache probe.",
)
|