Skip to content

GitLab

Unverified Commit 56eb5d0a authored by Xinyuan Tong's avatar Xinyuan Tong Committed by GitHub
Browse files

fix swa clear(): rename is_in_free_group to is_not_in_free_group (#9914)

parent 4ed9053e
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
python/sglang/srt/mem_cache/allocator.py
View file @ 56eb5d0a
...@@ -283,7 +283,7 @@ class SWATokenToKVPoolAllocator(BaseTokenToKVPoolAllocator): ...@@ -283,7 +283,7 @@ class SWATokenToKVPoolAllocator(BaseTokenToKVPoolAllocator):
self.swa_attn_allocator.clear() self.swa_attn_allocator.clear()
self.full_attn_allocator.clear() self.full_attn_allocator.clear()
self.full_to_swa_index_mapping.fill_(0) self.full_to_swa_index_mapping.fill_(0)
self.is_in_free_group = False self.is_not_in_free_group = True
self.free_group = [] self.free_group = []
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment