[Polish] Remove unused code

[horheynm]

SUMMARY:
load_dataset defined in test_generation.py is not used anywhere. Remove unused code

TEST PLAN:
Pass existing tests.
Use grep and search the repo with load_dataset, make sure its not from text_generation.py

[github-actions]

👋 Hi! Thank you for contributing to llm-compressor. Please add the ready label when the PR is ready for review.

Note: This is required to complete the testing suite, please only add the label once the PR is code complete and local testing has been performed.