Fix a number of type issues
Fix problems with some types. This enables Python's static type checks to correctly identify some issues before runtime.
Showing
- optimus/dataloader.py 5 additions, 5 deletionsoptimus/dataloader.py
- optimus/datasets/tinystories.py 3 additions, 3 deletionsoptimus/datasets/tinystories.py
- optimus/datasets/wikitext103.py 3 additions, 3 deletionsoptimus/datasets/wikitext103.py
- optimus/example_training.py 0 additions, 6 deletionsoptimus/example_training.py
- optimus/model.py 1 addition, 1 deletionoptimus/model.py
- optimus/tokenizer.py 4 additions, 4 deletionsoptimus/tokenizer.py
- optimus/trainer.py 5 additions, 4 deletionsoptimus/trainer.py
Loading
Please register or sign in to comment