apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-04 16:22:57 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

polish code with new pre-commit rule (#2923)

Browse Source
This commit is contained in:
Zero Rains
2025-07-19 23:19:27 +08:00
committed by GitHub
parent b8676d71a8
commit 25698d56d1
424 changed files with 14307 additions and 13518 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
scripts/extract_mtp_weight_from_safetensor.py
View File

@@ -28,19 +28,21 @@ from safetensors.numpy import save_file as safe_save_file
def parse_args():
""""""
parser = argparse.ArgumentParser(
description="Extract and save MTP weights from safetensors.")
parser.add_argument("-i",
"--input_dir",
type=str,
required=True,
help="Path to the input safetensors model directory.")
parser = argparse.ArgumentParser(description="Extract and save MTP weights from safetensors.")
parser.add_argument(
"-i",
"--input_dir",
type=str,
required=True,
help="Path to the input safetensors model directory.",
)
parser.add_argument(
"-o",
"--output_dir",
type=str,
required=True,
help="Path to the output directory for saving processed weights.")
help="Path to the output directory for saving processed weights.",
)
return parser.parse_args()