VISTA
VISTA copied to clipboard
prompt encoder wrong device
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).