Layers displayed re-rendering bug on different model heatmaps #62
Description
Generating a new heatmap on the same chart with a different model doesn't update the layers displayed properly, so it shows a truncated view if the first model was smaller than the new model used.
From gpt2 -> gpt-j
The heatmap still shows only 12 layers. Clicking the reset button updates with the right behavior.