Support bitwise weight checksum verifier (#16729)

This commit is contained in:
fzyzcjy
2026-01-08 20:19:25 +08:00
committed by GitHub
parent fb04e7e3c8
commit d54f0a10b4
4 changed files with 340 additions and 0 deletions

View File

@@ -422,6 +422,13 @@ class DefaultModelLoader(BaseModelLoader):
else:
hf_folder = model_name_or_path
server_args = get_global_server_args()
if server_args and server_args.model_checksum is not None:
from sglang.srt.utils.model_file_verifier import verify
checksums_source = server_args.model_checksum or model_name_or_path
verify(model_path=hf_folder, checksums_source=checksums_source)
hf_weights_files: List[str] = []
for pattern in allow_patterns:
hf_weights_files += glob.glob(os.path.join(hf_folder, pattern))