yya518/FinBERT

using CPU error

YeonGuKang opened this issue · 0 comments

first, I really appreciate this model.

In my environment CUDA is available, but if i run the model, the process will die since using only CPU.
Is there a separate setting method for using CUDA?

Thank you again for this model.