Fix: avoid AttributeError in attention_forward_hook when output[1] is…

[millerlight]

Problem

When the attention hook receives None as the second output, an AttributeError occurs.

Solution

Added a guard clause:

if output[1] is None: return