search

MaartenGr / KeyBERT

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

Loading local models #248

Open yhtest1 opened 1 month ago

yhtest1 commented 1 month ago

Is it possible to load local embeddings model instead of downloading from Huggingface?

MaartenGr commented 1 month ago

Yes, you can load any local model using SentenceTransformers and then pass that model directly to KeyBERT.