Converted pytorch_model.bin via utility in rust-bert using following command:

python3 utils/convert_model.py ../pytorch_model.bin

libtorch: libtorch-cxx11-abi-shared-with-deps-2.1.0+cu118
rust-bert commit/1f4d344668232da8e669e7fea1391c8829d5d1e3

BERT community org

cc @lysandre not sure if we should merge rust models in the repo directly?

BERT community org

IMO we can merge

BERT community org

Alright, let's merge!

lysandre changed pull request status to merged

Sign up or log in to comment