Support bitwise weight checksum verifier (#16729)
This commit is contained in:
@@ -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))
|
||||
|
||||
Reference in New Issue
Block a user