diff --git a/python/sglang/srt/managers/scheduler.py b/python/sglang/srt/managers/scheduler.py index 3ff6eaf57..7dd17793a 100644 --- a/python/sglang/srt/managers/scheduler.py +++ b/python/sglang/srt/managers/scheduler.py @@ -2333,9 +2333,8 @@ class Scheduler( def release_memory_occupation(self, recv_req: ReleaseMemoryOccupationReqInput): tags = recv_req.tags - import subprocess - if tags is None: + if tags is None or len(tags) == 0: tags = [GPU_MEMORY_TYPE_WEIGHTS, GPU_MEMORY_TYPE_KV_CACHE] if GPU_MEMORY_TYPE_KV_CACHE in tags: @@ -2353,6 +2352,7 @@ class Scheduler( def resume_memory_occupation(self, recv_req: ResumeMemoryOccupationReqInput): tags = recv_req.tags + if tags is None or len(tags) == 0: tags = [GPU_MEMORY_TYPE_WEIGHTS, GPU_MEMORY_TYPE_KV_CACHE]