OLMo icon indicating copy to clipboard operation
OLMo copied to clipboard

is_causal=attention_bias is None

Open nkkbr opened this issue 1 year ago • 1 comments

🐛 Describe the bug

https://github.com/allenai/OLMo/blob/5789cfe32390a0e80417e98285647cb8b41029ae/olmo/model.py#L598-L605

should the line 604 be is_causal=attention_bias is not None ?

Versions

Python 3.10.14

nkkbr avatar May 29 '24 00:05 nkkbr

Hey @nkkbr, when attention_bias is not None, the attention_bias includes a causal mask, so we don't to set is_causal to True since that would be redundant.

epwalsh avatar May 31 '24 16:05 epwalsh

Hi, thanks again for the inquiry! We’re currently working on closing out old tickets, so we’re closing this out for now, but if you require a follow-up response, please re-open and we will get back to you!

baileykuehl avatar Jul 01 '25 17:07 baileykuehl