Support sanity checking weight consistency especially for RL (#13854)

2025-11-27 20:25:12 +08:00
parent 2bc8ee8b74
commit 25758647b1
7 changed files with 156 additions and 0 deletions
@@ -71,6 +71,7 @@ from sglang.srt.managers.io_struct import (
    BaseReq,
    BatchTokenizedEmbeddingReqInput,
    BatchTokenizedGenerateReqInput,
+    CheckWeightsReqInput,
    ClearHiCacheReqInput,
    ClearHiCacheReqOutput,
    CloseSessionReqInput,
@@ -568,6 +569,7 @@ class Scheduler(
                (GetWeightsByNameReqInput, self.get_weights_by_name),
                (ReleaseMemoryOccupationReqInput, self.release_memory_occupation),
                (ResumeMemoryOccupationReqInput, self.resume_memory_occupation),
+                (CheckWeightsReqInput, self.check_weights),
                (SlowDownReqInput, self.slow_down),
                (ProfileReq, self.profile),
                (FreezeGCReq, self.handle_freeze_gc),