File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -335,11 +335,11 @@ def __init__(
335335 self .sparse_block_size = self .blocksparse_params .block_size
336336 self .head_sliding_step = self .blocksparse_params .head_sliding_step
337337
338- suppored_head_sizes = PagedAttention .get_supported_head_sizes ()
339- if head_size not in suppored_head_sizes :
338+ supported_head_sizes = PagedAttention .get_supported_head_sizes ()
339+ if head_size not in supported_head_sizes :
340340 raise ValueError (
341341 f"Head size { head_size } is not supported by PagedAttention. "
342- f"Supported head sizes are: { suppored_head_sizes } ." )
342+ f"Supported head sizes are: { supported_head_sizes } ." )
343343
344344 self .tp_size = get_tensor_model_parallel_world_size ()
345345 self .tp_rank = get_tensor_model_parallel_rank ()
You can’t perform that action at this time.
0 commit comments