Need a way to force load on CPU when an unsupported GPU throws a pytorch error.

dmmiller612 / bert-extractive-summarizer

Easy to use extractive text summarization with BERT

MIT License

1.39k stars 307 forks source link

Need a way to force load on CPU when an unsupported GPU throws a pytorch error. #152

Open michael-newsrx opened 1 year ago

michael-newsrx commented 1 year ago

I need a way to load on CPU when an unsupported GPU throws a pytorch error. This is a mixed project that is using both GPU and CPU modules and there is a need to run the sbertsummarizer on the cpu without conflicting with other modules using the GPU.