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

DistilBertModel support #163

Open
AbrJA opened this issue Jan 3, 2024 · 1 comment
Open

DistilBertModel support #163

AbrJA opened this issue Jan 3, 2024 · 1 comment

Comments

@AbrJA
Copy link

AbrJA commented Jan 3, 2024

Hi there,

First of all, thanks for the wonderful package you have done.

I'm trying to load a HuggingFace Model but I got these warnings and an error:

using Transformers
using Transformers.TextEncoders
using Transformers.HuggingFace

textenc, model = hgf"sentence-transformers/distiluse-base-multilingual-cased-v1"

┌ Warning: startsym <s> not in vocabulary, this might cause problem.
└ @ Transformers.TextEncoders [/home/ajaimes/.julia/packages/Transformers/lD5nW/src/textencoders/bert_textencoder.jl:77](https://file+.vscode-resource.vscode-cdn.net/home/ajaimes/.julia/packages/Transformers/lD5nW/src/textencoders/bert_textencoder.jl:77)

┌ Warning: endsym </s> not in vocabulary, this might cause problem.
└ @ Transformers.TextEncoders [/home/ajaimes/.julia/packages/Transformers/lD5nW/src/textencoders/bert_textencoder.jl:78](https://file+.vscode-resource.vscode-cdn.net/home/ajaimes/.julia/packages/Transformers/lD5nW/src/textencoders/bert_textencoder.jl:78)

Unknown model type: distilbert

Do you have in mind adding support for this type of models (distilbert)? Or is there some way to achieve this?

Thank you in advance!

@chengchingwen
Copy link
Owner

Haven't taken a close look but it shouldn't be too hard to add since it should be similar to bert. Though I couldn't do it right away, there is a small dev doc in the development branch about adding new models, hope that helped.

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

No branches or pull requests

2 participants