nt or list(int), optional, defaults to "auto") —
When "auto", input to the attention heads is halved, so attention is computed in two steps. If
"max", maximum amount of memory is saved by running only one slice at a time. If a number is
provided, uses as many slices as attention_head_dim // slice_size. In this case, attention_head_dim
must be a multiple of slice_size.    Enable sliced attention computation. When this option is enabled, the atte