Hz, Ji 50378cbf6c device agnostic models testing (#27146) ...

* device agnostic models testing

* add decorator `require_torch_fp16`

* make style

* apply review suggestion

* Oops, the fp16 decorator was misused

2023-10-31 18:12:14 +01:00

..

__init__.py

Add Mega: Moving Average Equipped Gated Attention (#21766)

2023-03-24 08:17:27 -04:00

test_modeling_mega.py

device agnostic models testing (#27146)

2023-10-31 18:12:14 +01:00