search

IBM / pytorch-large-model-support

Large Model Support in PyTorch
Apache License 2.0
131 stars 19 forks source link

LMS for Pytorch 1.6 #9

Open jplij opened 3 years ago

jplij commented 3 years ago

Hi guys, congratulations on the great work! I really like this LMS feature. I have seen that it has officially supported Pytorch 1.5. So, I want to ask when WML CE and LMS will add Pytorch 1.6 or even 1.7 version to its conda channel?

Thanks in advance!

kalliste commented 3 years ago

+1

Would be in to work on this if someone is available to give guidance

ghomasHudson commented 3 years ago

This would be super helpful - Pytorch 1.5 is getting too old for some downstream pytorch libraries now (e.g. lightning-transformers)

ghomasHudson commented 3 years ago

+1

Would be in to work on this if someone is available to give guidance

Have you just tried applying (and tweaking) the 1.5 patches?