updated _make_causal_mask and _expand_mask to newer transformer version efd5117 TaliDror commited on 13 days ago