search

MaartenGr / KeyBERT

Minimal keyword extraction with BERT
https://MaartenGr.github.io/KeyBERT/
MIT License
3.47k stars 344 forks source link

Cuda fix #181

Closed MaartenGr closed 1 year ago

MaartenGr commented 1 year ago

It seems that using pre-calculated embeddings throws an error since it expects embeddings to be a cuda tensor and not a numpy array.