* fix issue and add some tests * fix issue and add some tests * updated doc string gpt2
Attach a file by drag & drop or click to upload
