cerebras.modelzoo.tools#

checkpoint_converters

compat

convert_checkpoint

convert_config_to_mup

This script uses user provided SP GPT-2/3 yaml config to calculate and then generate a correspoding muP yaml config.

fold_lora_checkpoint