minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

Add adapter_model.safetensors to corruption validation for LoRA (#14022)

Browse Source
This commit is contained in:
alisonshao
2025-11-26 14:52:46 -08:00
committed by GitHub
parent 44d0a848da
commit a223402ffb
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/srt/model_loader/weight_utils.py
+6 -1
View File
@@ -393,7 +393,12 @@ def find_local_hf_snapshot_dir(
for f in local_weight_files: for f in local_weight_files:
base_name = os.path.basename(f) base_name = os.path.basename(f)
# Check if this is a single model file (not sharded) # Check if this is a single model file (not sharded)
if base_name in ["model.safetensors", "pytorch_model.safetensors"]: # Include adapter_model.safetensors for LoRA adapters
if base_name in [
"model.safetensors",
"pytorch_model.safetensors",
"adapter_model.safetensors",
]:
if not _validate_safetensors_file(f): if not _validate_safetensors_file(f):
logger.info( logger.info(
"Corrupted model file %s for %s. " "Corrupted model file %s for %s. "