simpleT5 icon indicating copy to clipboard operation
simpleT5 copied to clipboard
verified publisher icon Shivanandroy

TPU support

Open peregilk opened this issue 3 years ago • 4 comments

Any chance you can add TPU support in the Colab? I think this is supported more of less out of the box now in the newest pyTorch Lightning versions.

peregilk avatar Feb 02 '22 16:02 peregilk

This is what I was about to ask! Would be amazing.

swcrazyfan avatar Mar 07 '22 11:03 swcrazyfan

I'm working through the PyTorch Lightning docs, and I think I almost have working TPU support soon! I'll submit a pull request at that time.

swcrazyfan avatar Mar 08 '22 08:03 swcrazyfan

@swcrazyfan @Shivanandroy Any status on this?

peregilk avatar Aug 23 '22 12:08 peregilk

Yeah @Shivanandroy will you add this feature?

sr5434 avatar Mar 14 '23 15:03 sr5434