Skip to content

Fix inference with cpu only configs#9

Merged
wyli merged 1 commit intoProject-MONAI:mainfrom
adamaji:cpu-infer
Jul 20, 2023
Merged

Fix inference with cpu only configs#9
wyli merged 1 commit intoProject-MONAI:mainfrom
adamaji:cpu-infer

Conversation

@adamaji
Copy link
Copy Markdown
Contributor

@adamaji adamaji commented Jul 20, 2023

Description

Exciting work here! I noticed inference has some hard coded .cuda() calls. The "everything" segmentation isn't really feasible on CPU, and there seem to be some issues with multiple-prompt infers on my end, but this at least allows trying out single prompt infers with CPU only configs.

Types of changes

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

Copy link
Copy Markdown
Contributor

@wyli wyli left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks, it looks good to me.

@wyli wyli merged commit 573596a into Project-MONAI:main Jul 20, 2023
@adamaji adamaji deleted the cpu-infer branch July 20, 2023 12:18
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.

2 participants