attention_matrix&#39;, &#39;query_token_heatmap&#39;, &#39;key_token_heatmap&#39;, &#39;dimension_heatmap&#39;, &#39;token_dim_heatmap&#39;, &#39;line_grid&#39;`.</span>
<span class="sd">    color : (str, dict)</span>
<span class="sd">        A color to use for rendering components. Single color for all components or a dictionary of colors with</span>
<span class="sd">        (key: chart_type, value: color).</span>
<span class="sd">        If not provided, it defaults to &#39;blue&#39;.</span>
<span class="