fix:watchdog thread exception (#12328)
This commit is contained in:
@@ -2330,6 +2330,23 @@ class Scheduler(
|
|||||||
f"{swa_available_size=}, "
|
f"{swa_available_size=}, "
|
||||||
f"{swa_evictable_size=}, "
|
f"{swa_evictable_size=}, "
|
||||||
)
|
)
|
||||||
|
elif self.is_hybrid_gdn and isinstance(self.tree_cache, MambaRadixCache):
|
||||||
|
(
|
||||||
|
_,
|
||||||
|
_,
|
||||||
|
_,
|
||||||
|
_,
|
||||||
|
full_available_size,
|
||||||
|
full_evictable_size,
|
||||||
|
mamba_available_size,
|
||||||
|
mamba_evictable_size,
|
||||||
|
) = self._get_mamba_token_info()
|
||||||
|
info_msg = (
|
||||||
|
f"{full_available_size=}, "
|
||||||
|
f"{full_evictable_size=}, "
|
||||||
|
f"{mamba_available_size=}, "
|
||||||
|
f"{mamba_evictable_size=}, "
|
||||||
|
)
|
||||||
else:
|
else:
|
||||||
_, _, available_size, evictable_size = self._get_token_info()
|
_, _, available_size, evictable_size = self._get_token_info()
|
||||||
info_msg = f"{available_size=}, " f"{evictable_size=}, "
|
info_msg = f"{available_size=}, " f"{evictable_size=}, "
|
||||||
|
|||||||
Reference in New Issue
Block a user