replacement_library

Modules

modelopt.torch.puzzletron.replacement_library.build_replacement_library

This module constructs the replacement library JSON files from a puzzle directory containing multiple trained model checkpoints.

modelopt.torch.puzzletron.replacement_library.library

Replacement library for loading models with layer replacements (AnyModel / sharded HF checkpoints).

modelopt.torch.puzzletron.replacement_library.replacement_utils

This module provides helper functions for parsing, sorting, and analyzing layer replacement configurations used in the replacement library for model compression.

Replacement library for Puzzletron layer substitution.