setfit model on multiple GPUs?

huggingface / setfit

Efficient few-shot learning with Sentence Transformers

https://hf.co/docs/setfit

Apache License 2.0

2.24k stars 222 forks source link

setfit model on multiple GPUs? #541

Open twright8 opened 4 months ago

twright8 commented 4 months ago

Now that sentence transformers support multiple gpus, i was wondering if setfit supports it also? does Anyone has any idea about setting the number of GPUs in the setfittrainer

gxcuit commented 2 months ago

I have the same request