-
shakespeare.py - A Shakespeare like text generator built using a decoder transformer
-
shakespeare_ai.ipynb - The same project but in a Jupyter Notebook & extensively documented to be clear
You might want to scale down the hyperparameters as this model accounts to about 10 million parameters, I'd say don't run it on a CPU.