Allow for attention weights to be extracted.
#2
by
FJFehr
- opened
There is a small bug in indexing which didn't allow me to get the attentions out. I think this change fixes it.
There is a small bug in indexing which didn't allow me to get the attentions out. I think this change fixes it.