transformers/tests/models/deta
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 DETA (#20983) 2023-01-31 10:43:10 +01:00
test_image_processing_deta.py [`YOLOS`] Fix - return padded annotations (#29300) 2024-03-01 09:42:13 +00:00
test_modeling_deta.py Adding _tie_weights() to prediction heads to support low_cpu_mem_usage=True (#29024) 2024-05-07 11:12:21 +02:00