transformers/tests/models/wav2vec2
cyyever 786d9c5ed9
Fix more inefficient PT operations (#37060)
* Fix inefficient operations

* Remove cpu() call

* Reorder detach()

* Reorder detach()

* tolist without detach

* item without detach

* Update src/transformers/models/rag/modeling_rag.py

Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com>

* Update tests/models/encodec/test_modeling_encodec.py

Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com>

* Use detach().cpu().numpy

* Revert some numpy operations

* More fixes

---------

Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com>
2025-03-31 16:31:24 +01:00
..
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_feature_extraction_wav2vec2.py [tests] fix "Tester object has no attribute '_testMethodName'" (#34910) 2024-12-13 14:33:45 +01:00
test_modeling_flax_wav2vec2.py chore: fix typos in the tests directory (#36813) 2025-03-21 10:20:05 +01:00
test_modeling_tf_wav2vec2.py chore: fix typos in the tests directory (#36813) 2025-03-21 10:20:05 +01:00
test_modeling_wav2vec2.py Fix more inefficient PT operations (#37060) 2025-03-31 16:31:24 +01:00
test_processor_wav2vec2.py Support return_tensors in audio chat templates (#34601) 2025-03-25 11:08:47 +01:00
test_tokenization_wav2vec2.py Fix more inefficient PT operations (#37060) 2025-03-31 16:31:24 +01:00