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.

Sign up for GitHub

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

Jump to bottom

Support BERT-models #5

Merged

krishnap25 merged 1 commit into krishnap25:main from samhedin:bert-support

Mar 13, 2022

Contributor

samhedin commented Mar 3, 2022 •

edited

Context here: #3
I have run the code in the way suggested in the readme, but with roberta-large:

import mauve 

# call mauve.compute_mauve using raw text on GPU 0; each generation is truncated to 256 tokens
out = mauve.compute_mauve(p_text=p_text, q_text=q_text, device_id=0, max_text_length=256, verbose=False, featurize_model_name="roberta_large")
print(out.mauve)

I ran it once on 2-3 samples, and it did not crash, so, great success?


          Support BERT-models

24507c0

krishnap25 merged commit 3994fb2 into krishnap25:main

Owner

krishnap25 commented Mar 13, 2022

Looks good to me. Thanks!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment