tests/ctrl/test_modeling_ctrl.py · c04619ecf3ce1975d2716ff77e29e2875d9ddd60 · chenpangpang / transformers

Enable more test_torchscript (#16679) · c04619ec

Yih-Dar authored Apr 11, 2022



* update _create_and_check_torchscript

* Enable test_torchscript

* clear_class_registry
Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>

c04619ec

test_modeling_ctrl.py 8.15 KB

Replace test_modeling_ctrl.py