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

textEmbed default layers argument does not work with model = 'distilroberta-base' #35

Closed
sebsilas opened this issue Dec 20, 2022 · 1 comment

Comments

@sebsilas
Copy link

sebsilas commented Dec 20, 2022

The following does not work for the distilroberta-base model class.

test <- text::textEmbed("hi there", model = 'distilroberta-base')

But if you explicitly call the second-to-last layer, rather than using the -2 default, (here, layer 5), it works:

test2 <- text::textEmbed("hi there", model = 'distilroberta-base', layers = 5)
@sebsilas sebsilas changed the title textEmbed default layer argument does not work with model = 'distilroberta-base' textEmbed default layers argument does not work with model = 'distilroberta-base' Dec 20, 2022
@OscarKjell
Copy link
Owner

Thank you for reporting this. Please note that this should work with new versions of the text package now (currently need to download it from github; version .9.99.7 and above).

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