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

2025-11-26 14:52:46 -08:00
parent 44d0a848da
commit a223402ffb
1 changed files with 6 additions and 1 deletions
--- a/python/sglang/srt/model_loader/weight_utils.py
+++ b/python/sglang/srt/model_loader/weight_utils.py
@@ -393,7 +393,12 @@ def find_local_hf_snapshot_dir(
        for f in local_weight_files:
            base_name = os.path.basename(f)
            # 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):
                    logger.info(
                        "Corrupted model file %s for %s. "