tests/models/mega/test_modeling_mega.py · b8def689346c45958268ec389ee6242bddc6d78c · chenpangpang / transformers

Tanay Mehta authored Sep 06, 2023

* add: potential fix to mega chunking in decoder only model bug

* add: decoder with chunking test

* add: input_mask passed with input_ids

b8def689

test_modeling_mega.py 28.6 KB