Including DeBertaV2 for zero-shot

guillaume-be / rust-bert

Rust native ready-to-use NLP pipelines and transformer-based models (BERT, DistilBERT, GPT2,...)

https://docs.rs/crate/rust-bert

Apache License 2.0

2.51k stars 211 forks source link

Including DeBertaV2 for zero-shot #418

Closed jondot closed 9 months ago

jondot commented 10 months ago

Just followed the structure of DeBerta (non-v2), hopefully this makes sense? Model runs and gives me results that I agree with. The code was there, it just feels it wasn't rigged into the various enums that were needed