transformers/tests/models/deta
JB (Don) 725f4ad1cc
Add tie_weights() to LM heads and set bias in set_output_embeddings() (#28948)
* Add tie_weights() to LM heads and set bias in set_output_embeddings()

The bias were not tied correctly in some LM heads, and this change should fix that.

* Moving test_save_and_load_low_cpu_mem_usage to ModelTesterMixin

* Adding _tie_weights() to MPNet and Vilt

* Skip test for low cpu mem usage for Deta/DeformableDetr since they cannot init on meta device

* Rename to test name to save_load to match the convention
2024-02-14 20:39:01 +00:00
..
__init__.py Add DETA (#20983) 2023-01-31 10:43:10 +01:00
test_image_processing_deta.py [DETR] Update the processing to adapt masks & bboxes to reflect padding (#28363) 2024-02-13 18:27:06 +00:00
test_modeling_deta.py Add tie_weights() to LM heads and set bias in set_output_embeddings() (#28948) 2024-02-14 20:39:01 +00:00