Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Efficient KeyLLM + KeyBERT - Torch not compiled with CUDA enabled #207

Open
sam-frampton opened this issue Feb 11, 2024 · 1 comment
Open

Efficient KeyLLM + KeyBERT - Torch not compiled with CUDA enabled #207

sam-frampton opened this issue Feb 11, 2024 · 1 comment

Comments

@sam-frampton
Copy link
Contributor

sam-frampton commented Feb 11, 2024 •
edited
Loading

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
Copy link
Owner

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@MaartenGr @sam-frampton