LWM icon indicating copy to clipboard operation
LWM copied to clipboard
verified publisher icon LargeWorldModel

LWM-Chat in PyTorch

Open j-min opened this issue 1 year ago • 1 comments

Hi, thanks for releasing the code. Looks pretty interesting! I noticed that the LWM-Chat (multimodal) model checkpoint is only released in Jax. It would be great if you could release the model in PyTorch as well as you did for other text-only models!

j-min avatar Feb 14 '24 16:02 j-min

Hi, thanks for your interest! A pytorch version is on the roadmap, but may take a some time since both of us are rather occupied with other things at the moment.

wilson1yan avatar Feb 14 '24 23:02 wilson1yan