search

Shivanandroy / KeyPhraseTransformer

KeyPhraseTransformer lets you quickly extract key phrases, topics, themes from your text data with T5 transformer | Keyphrase extraction | Keyword extraction
MIT License
97 stars 13 forks source link

GPU question #4

Open sersoage opened 2 years ago

sersoage commented 2 years ago

Is it possible to run on GPU, if yes how? Thanks!

Shivanandroy commented 2 years ago

Hey @sersoage : In the current version it only runs on CPU. I will add a flag to run GPU or CPU in the next release.

For now, you can fork it and then make changes in predict method to run it on GPU: https://github.com/Shivanandroy/KeyPhraseTransformer/blob/80e2c8d7c4869d1e120f6e8004630812443e766b/keyphrasetransformer/keyphrasetransformer.py#L72

kunalgoyal9 commented 3 months ago

https://github.com/Shivanandroy/KeyPhraseTransformer/pull/11