镜像站点 · 本页由第三方 GitHub 只读镜像提供,非 GitHub 官方站点,不接受任何登录或凭据输入。前往 github.com
Skip to content

prompt encoder wrong device - #18

Open
elitap wants to merge 7 commits into
Project-MONAI:mainfrom
elitap:17_prompt_encoder_wrond_device
Open

elitap wants to merge 7 commits into
Project-MONAI:mainfrom
elitap:17_prompt_encoder_wrond_device

Conversation

@elitap

@elitap elitap commented Aug 30, 2023 •

Copy link
Copy Markdown

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

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

elitap and others added 7 commits August 23, 2023 17:33
@elitap elitap changed the title 17 prompt encoder wrond device 17 prompt encoder wrong device Aug 30, 2023
@elitap elitap changed the title 17 prompt encoder wrong device prompt encoder wrong device Aug 30, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Input of promt Encoder on wrong device (CPU)

1 participant