Efficient KeyLLM + KeyBERT - Torch not compiled with CUDA enabled

MaartenGr / KeyBERT

Minimal keyword extraction with BERT

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

MIT License

3.46k stars 344 forks source link

Open sam-frampton opened 7 months ago

sam-frampton commented 7 months ago

Testing out Efficient KeyLLM + KeyBERT in a pipeline, when the code hits the refinement stage using LLM I get error

"Error while extracting keywords from text: Torch not compiled with CUDA enabled"

This line that breaks the code inside extract keywords class - doc_embeddings = torch.from_numpy(doc_embeddings).float().to("cpu")

MaartenGr commented 7 months ago

Thanks for sharing this. Could you share your full code including the full error message?