tests/models/speecht5/test_modeling_speecht5.py · 7e1c7dc8b6989d96e7c35781bccb09468b49fc5e · chenpangpang / transformers · GitLab

Find file Blame History Permalink

Fix SpeechT5 `decoder_attention_mask` shape (#28071) · 7e1c7dc8
Yoach Lacombe authored Jun 14, 2024
```
* Fix SpeechT5

* add test foward with labels and attention mask

* make style
```
7e1c7dc8

test_modeling_speecht5.py 78.8 KB