Keep compression-related configuration flags in sync.
(cfg: Any)
| 276 | |
| 277 | |
| 278 | def synchronize_compression_config(cfg: Any) -> None: |
| 279 | """Keep compression-related configuration flags in sync.""" |
| 280 | |
| 281 | comp_cfg = cfg.compression_sim_cfg |
| 282 | |
| 283 | if getattr(cfg, "compression_sim", False): |
| 284 | comp_cfg.enabled = True |
| 285 | else: |
| 286 | cfg.compression_sim = comp_cfg.enabled |
| 287 | |
| 288 | if getattr(cfg, "entropy_model_opt", False): |
| 289 | comp_cfg.entropy.enabled = True |
| 290 | else: |
| 291 | cfg.entropy_model_opt = comp_cfg.entropy.enabled |
| 292 | |
| 293 | comp_cfg.entropy.model_type = getattr(cfg, "entropy_model_type", None) or comp_cfg.entropy.model_type |
| 294 | cfg.entropy_model_type = comp_cfg.entropy.model_type |
| 295 | |
| 296 | entropy_steps = getattr(cfg, "entropy_steps", None) |
| 297 | if entropy_steps: |
| 298 | comp_cfg.entropy.steps.update(entropy_steps) |
| 299 | cfg.entropy_steps = comp_cfg.entropy.steps |
| 300 | |
| 301 | if getattr(cfg, "shN_ada_mask_opt", False): |
| 302 | comp_cfg.mask.enabled = True |
| 303 | else: |
| 304 | cfg.shN_ada_mask_opt = comp_cfg.mask.enabled |
| 305 | |
| 306 | strategy = getattr(cfg, "shN_ada_mask_strategy", None) |
| 307 | if strategy is not None: |
| 308 | comp_cfg.mask.strategy = strategy |
| 309 | elif comp_cfg.mask.strategy is not None: |
| 310 | cfg.shN_ada_mask_strategy = comp_cfg.mask.strategy |
| 311 | |
| 312 | mask_steps = getattr(cfg, "ada_mask_steps", None) |
| 313 | if mask_steps is not None: |
| 314 | comp_cfg.mask.start_step = mask_steps |
| 315 | elif comp_cfg.mask.start_step is not None: |
| 316 | cfg.ada_mask_steps = comp_cfg.mask.start_step |
| 317 | |
| 318 | cfg.compression_sim_cfg = comp_cfg |