galai icon indicating copy to clipboard operation
galai copied to clipboard

Can be used on CPU?

Open SalvatoreRa opened this issue 3 years ago • 2 comments

Hi,

amazing works and very curious to try

can be used on a CPU (or parallelized on different CPUs)?

thank you,

Salvatore

SalvatoreRa avatar Nov 18 '22 13:11 SalvatoreRa

Hi Salvatore. Yes it is possible using the Hugging Face integration. The Readme has been uodated with the details.

Tugees avatar Nov 19 '22 17:11 Tugees

Thank you, I have just seen and definitely curious to try

SalvatoreRa avatar Nov 21 '22 09:11 SalvatoreRa

Hi @SalvatoreRa, as of galai version 1.1.0 you should be able to easily run the inference on CPU with load_model(..., num_gpus=0) (the default if there's no CUDA device). Please reopen if you still have any issues.

mkardas avatar Dec 09 '22 11:12 mkardas