Fix some issues with the wikitext103 dataset

Fix the wikitext103 dataset. This comes after considering a couple of things when reviewing TinyStories PR.

requested review from @vlad_andrei.badoiu1

assigned to @agherghescu2411

mentioned in commit 39950c13

merged