Skip to content

Commit d200151

Browse files
ch-wanjianan-gu
authored andcommitted
[DeepEP] fix: import buffer error (sgl-project#5179)
1 parent a47d09b commit d200151

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

python/sglang/srt/layers/moe/ep_moe/token_dispatcher.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ class DeepEPDispatchMode(IntEnum):
2828

2929
class DeepEPBuffer:
3030

31-
_buffer: Optional[Buffer] = None
31+
_buffer = None
3232
_dispatch_mode: Optional[DeepEPDispatchMode] = None
3333
_hidden_size: Optional[int] = None
3434
_num_max_dispatch_tokens_per_rank: Optional[int] = None
@@ -165,7 +165,7 @@ def combine_a(
165165
def combine_b(self, *args, **kwargs):
166166
raise NotImplementedError
167167

168-
def _get_buffer(self) -> Buffer:
168+
def _get_buffer(self):
169169
raise NotImplementedError
170170

171171

0 commit comments

Comments
 (0)