transformers/tests/models/deformable_detr
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 Add Deformable DETR (#17281) 2022-09-14 11:45:21 +02:00
test_image_processing_deformable_detr.py [`YOLOS`] Fix - return padded annotations (#29300) 2024-03-01 09:42:13 +00:00
test_modeling_deformable_detr.py Adding _tie_weights() to prediction heads to support low_cpu_mem_usage=True (#29024) 2024-05-07 11:12:21 +02:00