transformers/tests/models/jukebox
Joao Gante 248d5d23a2
Tests: replace `torch.testing.assert_allclose` by `torch.testing.assert_close` (#29915)
* replace torch.testing.assert_allclose by torch.testing.assert_close

* missing atol rtol
2024-03-28 09:53:31 +00:00
..
__init__.py Add Jukebox model (replaces #16875) (#17826) 2022-11-10 21:05:27 +01:00
test_modeling_jukebox.py Tests: replace `torch.testing.assert_allclose` by `torch.testing.assert_close` (#29915) 2024-03-28 09:53:31 +00:00
test_tokenization_jukebox.py Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01:00