search

ramsrigouthamg / Questgen.ai

Question generation using state-of-the-art Natural Language Processing algorithms
https://questgen.ai/
MIT License
910 stars 294 forks source link

Questgen supports GPU? #18

Closed pqrz closed 2 years ago

pqrz commented 3 years ago

Hi Team,

Questgen is very useful library for question generation.

We are able to use it well on CPU, but it consumes high runtime on large datasets.

Does Questgen also gives GPU support?

ramsrigouthamg commented 2 years ago

Hi @pqrz ,

The models are built over Huggingface itself so support GPU as well. Although you might need slight changes to add device as GPU in the text decoding code.