transformers/tests/models/rembert
Katie Le cc1d0685b3
Wrap RemBert integration test forward passes with torch.no_grad() (#21503)
added with torch.no_grad() to the integration tests and applied make style

Co-authored-by: Bibi <Bibi@katies-mac.local>
2023-02-08 14:00:52 +01:00
..
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_modeling_rembert.py Wrap RemBert integration test forward passes with torch.no_grad() (#21503) 2023-02-08 14:00:52 +01:00
test_modeling_tf_rembert.py Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00