VISTA icon indicating copy to clipboard operation
VISTA copied to clipboard

prompt encoder wrong device

Open elitap opened this issue 2 years ago • 0 comments

Fixes #17

Description

monailabel/utils/others/generic.device_map() returns "cuda:x" in case of multiple gpus, accordingly, the compare with device == "cuda" fails in monaivista....inferer.py and monaivista....utils.py. As a result the inputs for the promt encoder are never moved to "gpu"

Types of changes

  • [x] Non-breaking change (fix or new feature that would not break existing functionality).

elitap avatar Aug 30 '23 14:08 elitap