Colab notebook used to work fine, now breaks on "import optax" with error "No module named 'optax'". (Yes, this is the cell that says "Sometimes the next step errors for some reason, just run it again ¯_(ツ)_/¯", but repeated attempts don't work. This used to work every time).
If I keep on !pip installing every library that it can't find, it still fails when it gets to "from mesh_transformer.checkpoint import read_ckpt_lowmem" with the error "Cannot subclass <class 'typing._SpecialForm'>".
Colab notebook used to work fine, now breaks on "import optax" with error "No module named 'optax'". (Yes, this is the cell that says "Sometimes the next step errors for some reason, just run it again ¯_(ツ)_/¯", but repeated attempts don't work. This used to work every time).
If I keep on !pip installing every library that it can't find, it still fails when it gets to "from mesh_transformer.checkpoint import read_ckpt_lowmem" with the error "Cannot subclass <class 'typing._SpecialForm'>".