search

allenai / rainbow

Neural models of common sense. 🤖
Apache License 2.0
94 stars 6 forks source link

Convert Unicorn weights to torch #55

Closed ZhaozwTD closed 2 years ago

ZhaozwTD commented 2 years ago

Is there a way to convert Unicorn weights to torch?

Thanks for your nice codes!

nalourie-ai2 commented 2 years ago

Hi! I never implemented code to convert the weights to torch. The weights we released are in the original format used by T5, so any script for cross-loading T5's original weights into torch should also work here.