proteinbertからEmbeddingを取得する方法について


タンパク質言語モデルであるproteinbert<https://github.com/nadavbra/protein_bert>  
を用いて入力したアミノ酸配列のEmbeddingを取得する方法が知りたいです。  
fine-tuningを行わずに取得したEmbeddingを学習に用いたいです。  
詳しい方、よろしければご教授ください。

