[Reference](https://github.com/devalab/molgpt/blob/main/train/trainer.py#L169) Leaving it to ```None``` gets me errors in [this section](https://github.com/devalab/molgpt/blob/main/train/model.py#L198). If I ignore the ```assert``` statement, the model would fail in [this block](https://github.com/devalab/molgpt/blob/main/train/model.py#L206).
Reference
Leaving it to
Nonegets me errors in this section.If I ignore the
assertstatement, the model would fail in this block.