LennardZuendorf's picture
fix: fixing attention visualization
c7e16d0