transformers/templates/adding_a_new_model/tests
Sylvain Gugger d951c14ae4
Model output test (#6155)
* Use return_dict=True in all tests

* Formatting
2020-07-31 09:44:37 -04:00
..
test_modeling_tf_xxx.py GPU text generation: mMoved the encoded_prompt to correct device 2020-01-06 15:11:12 +01:00
test_modeling_xxx.py Model output test (#6155) 2020-07-31 09:44:37 -04:00
test_tokenization_xxx.py [docs] Add integration test example to copy pasta template (#5961) 2020-07-22 12:48:38 -04:00