candle icon indicating copy to clipboard operation
candle copied to clipboard

Published 20 hours ago • verified publisher icon huggingface

How to get all layers attentions?

Open kitty-eu-org opened this issue 1 year ago • 0 comments

I only see that candle returns last_hidden_state, but not all_hidden_states and attentions. I want to get attentions. Can I submit a PR to do this? I originally wanted to define the Model myself, but I found that all its methods are private

kitty-eu-org avatar Jul 02 '24 02:07 kitty-eu-org