Update Jukebox tests
# What does this PR do?

Updage Jukebox tests, for PyTorch 2.0.

The reason is the same as in #21975: tiny diff in scores, but sampling can give different results.
_The documentation is not available anymore as the PR was closed or merged._<|||||>test failure irrelevant to this PR.