Endless loop with multiple GPU ?
#34
by
Grosper
- opened
I have 2 GPUs and BGEM3FlagModel seems to be having an issue (infinite loop) in my case (model.encode never finishes). I have solved the problem with BGEM3FlagModel(..., device="cuda:0",...)`.
Regards.