setu4993 / convert-labse-tf-pt

Convert LaBSE model from TF Hub to PyTorch.
https://huggingface.co/setu4993/LaBSE
MIT License
14 stars 0 forks source link

Will it work for LaBSE 2 version? #1

Closed takiholadi closed 2 years ago

takiholadi commented 2 years ago

https://tfhub.dev/google/LaBSE/2

setu4993 commented 2 years ago

@takiholadi : Thanks for flagging! Haven't tried it, but I'm assuming yes. I can try it over the weekend and check in a few days.

setu4993 commented 2 years ago

@takiholadi : Updated the model to v2 on the HF Hub. FWIW, it is an identical model (see this test) with slightly different inputs.