Fix flax GPT-J-6B linking model in tests
# What does this PR do?

Fixes the loading of the model `EleutherAI/gpt-j-6B` as the current code links `EleutherAI/gptj-6B` which does not exist and ends up failing the test.
_The documentation is not available anymore as the PR was closed or merged._