transformers/tests/models/encodec
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 [WIP] add EnCodec model (#23655) 2023-06-14 18:57:23 +02:00
test_feature_extraction_encodec.py fix: Replace deprecated `assertEquals` with `assertEqual` (#30241) 2024-04-15 09:36:06 +01:00
test_modeling_encodec.py Adding _tie_weights() to prediction heads to support low_cpu_mem_usage=True (#29024) 2024-05-07 11:12:21 +02:00