transformers/tests/models/marian
JB (Don) 54a2361a29
Adding _tie_weights() to prediction heads to support low_cpu_mem_usage=True (#29024)
* Adding _tie_weights() to prediction heads to support low_cpu_mem_usage=True

* Testing for the non-safe-tensors case, since the default is safe-tensors already

* Running fixup/fix-copies

* Adding accelerate annotations to tests
2024-05-07 11:12:21 +02:00
..
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_modeling_flax_marian.py [`Tokenizer`] Fix slow and fast serialization (#26570) 2023-10-18 16:30:53 +02:00
test_modeling_marian.py Adding _tie_weights() to prediction heads to support low_cpu_mem_usage=True (#29024) 2024-05-07 11:12:21 +02:00
test_modeling_tf_marian.py [`Tokenizer`] Fix slow and fast serialization (#26570) 2023-10-18 16:30:53 +02:00
test_tokenization_marian.py Adds pretrained IDs directly in the tests (#29534) 2024-03-13 14:53:27 +01:00