DogsHead

Results 2 issues of DogsHead

running show_cross_attention(controller, 16, ["up", "down"]), it throws KeyError, what's the problem