naver / biobert-pretrained

BioBERT: a pre-trained biomedical language representation model for biomedical text mining
667 stars 88 forks source link

Using HuggingFace transformers library #28

Closed lizmonch closed 8 months ago

lizmonch commented 8 months ago

I am trying to fine tune the BioBERT model using free-text laboratory data in a data safe haven. The biobert-pretrained model was downloaded into a local directory.

I tried to load the model weights using the following code: From transformers import TFAutoModelForSequenceClassification model = TFAutoModelForSequenceClassification.from_pretrained('/localdirectorypath')

I get an error message that no file named tf_model.h5 or pytorch_model.bin is found in the local directory.

Am I not able to use the HuggingFace transformers library with the biobert-pretrained model?