search

sensein / b2aiprep

Apache License 2.0
5 stars 6 forks source link

add device support #9

Closed satra closed 5 months ago

satra commented 5 months ago

closes #8 - adds support for selecting device to run huggingface models on. this does not add device support to all calls, only the hparams based functions.