class="sd">        List of tokens</span>
<span class="sd">    values : (ArrayLike or dict[str, ArrayLike])</span>
<span class="sd">        Values to visualize. (key: name, value: list of values with shape [num_tokens])</span>
<span class="sd">    token_info : Optional[list[str]]</span>
<span class="sd">        Aditional info about the tokens. Shape [num_tokens]</span>
<span class="sd">    remove_padding : bool</span>
<span class="sd">        Whether to remove padding in the visualization</span>
<span class=