Hi, kindly check the following merge commit and test it out if it will resolve your issue:

https://github.com/nshepperd/gpt-2/pull/6

Alternatively, you can post-process it to trim off any data after the delimiter.