Simmonstt's picture
Fix: Add missing torch import in generate_attention_visualization
72100db verified