Can not run the code Transformers learn in-context by gradient descent/constructed_token_setup.ipynb

[pkumath]

The error as following

[JosephSBoyle]

Confirming that I get an error (albeit a different error message) on the same import statement.