maxtext.experimental.agent.ckpt_conversion_agent.ground_truth.llama3 module#
Lllama3 ckpt conversation agent ground truth hook functions.
- maxtext.experimental.agent.ckpt_conversion_agent.ground_truth.llama3.LLAMA31_MAXTEXT_TO_HF_PARAM_HOOK_FN(config, scan_layers=False, saving_to_hf=False)[source]#
Creates parameter transformation functions for converting between MaxText and HuggingFace formats. This function generates a mapping of transformation functions that handle the necessary conversions between MaxText and HuggingFace parameter formats, including operations like reshaping.