A question about casual tracing code

[Zijian007]

Thank you for your great work. I notice that during the casual tracing, when the kind is None, the “trace_important_states ” works layer by layer， but when the kind is attention or mlp, the “trace_important_window” works in layer windows. Why the layer range is different when the kind changes.