Loading local models - Githubissues

MaartenGr / KeyBERT

Minimal keyword extraction with BERT

https://MaartenGr.github.io/KeyBERT/

MIT License

3.55k stars 351 forks source link

Open yhtest1 opened 2 months ago

yhtest1 commented 2 months ago

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

MaartenGr commented 2 months ago

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