You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I'm currently adding Prism to Jury. Current implementation forces using GPU if cuda.is_available() returns True; however, this behavior is problematic as a user is unable to use CPU e.g if VRAM is not sufficient for loading the underlying prism model. It'd be very good if use_cuda is exposed as an option to user, so the user can decide whether to use CPU/GPU.
I'm currently adding Prism to Jury. Current implementation forces using GPU if
cuda.is_available()
returns True; however, this behavior is problematic as a user is unable to use CPU e.g if VRAM is not sufficient for loading the underlying prism model. It'd be very good ifuse_cuda
is exposed as an option to user, so the user can decide whether to use CPU/GPU.Refer to note on obss/jury#79.
The text was updated successfully, but these errors were encountered: