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