RWKV-LM
RWKV-LM copied to clipboard
Fix missing return in __getitem__ for uint16 in dataset.py
The getitem method did not return any value when args.data_type == "uint16", causing data loaders to receive None. Added an explicit return x, y to match the behavior of other data types.