- [x] the data loaders for training - [x] the data loaders for `generating` - [x] the `generate.py` script - [ ] ?