Get attention weights of input tokens

#39
by Reynier - opened

Could someone help me obtain the attention weights that the model gives to the input tokens?

Sign up or log in to comment