Endless loop with multiple GPU ?

#34

by Grosper - opened Mar 28

Discussion

Grosper

Mar 28

•

edited Mar 28

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.

Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.

Tap or paste here to upload images

· Sign up or log in to comment