comment

hengtaoguo · hengtaoguo · commit fba143fcf103 · 2026-05-08T20:43:53.000Z
diff --git a/src/maxtext/checkpoint_conversion/to_huggingface.py b/src/maxtext/checkpoint_conversion/to_huggingface.py
@@ -119,6 +119,7 @@ def _get_model_mappings(
     ValueError: If mappings for the specified `model_name` are not found.
   """
   target_model_name = model_name
+  # Fall back to base model mappings if the given -Instruct variant belongs to the same family.
   if target_model_name not in PARAM_MAPPING and target_model_name.endswith("-Instruct"):
     base_name = target_model_name[: -len("-Instruct")]
     if base_name in HF_IDS and base_name in PARAM_MAPPING:
