File size: 2,045 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 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 | from __future__ import annotations
def add_worldcache_args(parser) -> None:
parser.add_argument(
"--use_worldcache",
action="store_true",
help="Enable WorldCache v1 acceleration for Ctrl-World.",
)
parser.add_argument(
"--worldcache_rel_l1_thresh",
type=float,
default=0.03,
help="Base relative L1 threshold for WorldCache skip decisions.",
)
parser.add_argument(
"--worldcache_ret_ratio",
type=float,
default=0.4,
help="Fraction of denoising steps to keep dense before WorldCache becomes eligible.",
)
parser.add_argument(
"--worldcache_probe_depth",
type=int,
default=3,
help="Number of attention-bearing UNet stages used for the WorldCache probe.",
)
parser.add_argument(
"--worldcache_motion_sensitivity",
type=float,
default=5.0,
help="Velocity sensitivity for dynamic WorldCache thresholding.",
)
parser.add_argument(
"--worldcache_hf_enabled",
action="store_true",
help="Enable high-frequency drift guard for WorldCache.",
)
parser.add_argument(
"--worldcache_hf_thresh",
type=float,
default=0.01,
help="High-frequency drift threshold for WorldCache.",
)
parser.add_argument(
"--worldcache_saliency_enabled",
action="store_true",
help="Enable saliency-weighted probe drift for WorldCache.",
)
parser.add_argument(
"--worldcache_saliency_weight",
type=float,
default=5.0,
help="Saliency weighting factor for WorldCache probe drift.",
)
parser.add_argument(
"--worldcache_osi_enabled",
action="store_true",
help="Enable optimal scalar interpolation gamma for WorldCache output extrapolation.",
)
parser.add_argument(
"--worldcache_dynamic_decay",
action="store_true",
help="Relax the WorldCache threshold toward late denoising steps.",
)
|