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.
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.