Sorry about the hassle loading the weights directly. But I actually have some bonus material on that to load them from the HF model hub as an alternative (should be referenced in the book itself):

→ View original post on X — @rasbt