Add patch to fix when attention mask is None in task_heads.py 4bdfa1e kokolamba commited on Sep 30, 2025