search

LxMLS / lxmls-toolkit

Machine Learning applied to Natural Language Processing Toolkit used in the Lisbon Machine Learning Summer School
Other
222 stars 216 forks source link

Unified attention #199

Open israfelsr opened 1 year ago

israfelsr commented 1 year ago

Created function that transfer weights of pretrained models to the new CasualAttention module. Tested with some prompts and working perfectly 🤗. Will add it to student branch.