tracel-ai / models

Models and examples built with Burn
Apache License 2.0
180 stars 24 forks source link

Add LM head and masked fill_mask for bert-burn. #34

Closed seurimas closed 5 months ago

seurimas commented 5 months ago

I'm not sure if this matches the structure desired for this crate, but I had figured out masked token inference for a personal project and it seemed like a waste not to try and contribute. Please let me know what sort of changes might be needed to get this merged, if this is something worth merging.

Additionally, this fixed #30.