We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 1a8792d commit b19d8f0Copy full SHA for b19d8f0
1 file changed
python/sgl_jax/srt/mem_cache/recurrent_state_pool.py
@@ -83,16 +83,6 @@ def __init__(
83
# total_slots: size+1 (for dummy slot 0), ceil to dp_size
84
self.total_slots = _ceil_to(size + 1, dp_size)
85
86
- assert size > 0, f"size must be > 0, got {size}"
87
- assert num_heads > 0
88
- assert head_dim > 0
89
- assert num_k_heads > 0
90
- assert head_k_dim > 0
91
- assert (
92
- conv_kernel_size >= 2
93
- ), f"conv_kernel_size must be >= 2 (got {conv_kernel_size}); K=1 produces empty conv buffers."
94
- assert self.proj_size > 0
95
-
96
self.mesh = mesh
97
self.recurrent_partition_axis = recurrent_partition_axis
98
self.conv_partition_axis = conv_partition_axis
0 commit comments