search

pdrm83 / sent2vec

How to encode sentences in a high-dimensional vector space, a.k.a., sentence embedding.
MIT License
133 stars 12 forks source link

fixed issues with GPU and SpaCy #12

Closed almarengo closed 2 years ago

almarengo commented 2 years ago

Added option in class Vectorizer to have tensor and model to GPU, if available Fixed SpaCy package name when loading and deprecated error. Added exception with some instructions how to install "en_core_web_md" if package not found

pdrm83 commented 2 years ago

Thank you. I don't have GPU on my laptop. Did you find a chance to test it as well?

pdrm83 commented 2 years ago

BTW. If you believe this PR will resolve the issue raised by the community please kindly close the issue after this PR get merged.

almarengo commented 2 years ago

I have a GPU on my laptop and I tested it. It works.

This PR will resolve two issues raised by the community. I will close those once this PR is merged.

almarengo commented 2 years ago

Thanks for closing the PR. I just realized that I could have linked the issues with the PR :( Now I'm not able to close the issues but you should be. The solved issues are #8 and #10