def main()

in src/hyperpod_nemo_adapter/scripts/merge_peft_checkpoint.py [0:0]

• 22 lines of code
• 1 McCabe index (conditional complexity)

def main():
    parser = argparse.ArgumentParser(
        description="Script for merging a Hugging Face model with a PEFT adapter checkpoint"
    )

    parser.add_argument(
        "--hf_model_name_or_path",
        type=str,
        required=True,
        help="The Hugging Face model name or path to load the model from.",
    )
    parser.add_argument(
        "--peft_adapter_checkpoint_path", type=str, required=True, help="Path to the PEFT adapter checkpoint."
    )
    parser.add_argument(
        "--output_model_path", type=str, required=True, help="Path where the merged model will be saved."
    )
    parser.add_argument(
        "--hf_access_token", type=str, default=None, help="Optional Hugging Face access token for authentication."
    )
    parser.add_argument("--deepseek_v3", type=bool, default=False, help="Whether the model is DeepSeek V3 model.")

    args = parser.parse_args()
    run(args)