Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

PyTorch-Transformers: add possibility to pass model path to embeddings #993

Merged
merged 3 commits into from
Aug 14, 2019

Conversation

stefan-it
Copy link
Member

Hi,

this PR adds the possibility to pass a model path to the PyTorch-Transformer based embeddings.

This is highly needed when Flair users want to use own trained models, like e.g. an own trained XLNet model.

The argument name is changed from model to pretrained_model_name_or_path (as it is named in pytorch-transformers).

@stefan-it
Copy link
Member Author

Btw: an extra section that shows how to use an own trained model is coming very soon (it shows how to use sciBERT with flair) 😅

@alanakbik
Copy link
Collaborator

Awesome, thanks for adding this!

@alanakbik
Copy link
Collaborator

👍

1 similar comment
@yosipk
Copy link
Collaborator

yosipk commented Aug 14, 2019

👍

@yosipk yosipk merged commit abc74ab into master Aug 14, 2019
@stefan-it stefan-it deleted the flair-pytorch-transformers-own-models branch August 14, 2019 11:44
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants