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