Handle CUDA NMS fallback in YOLO11 plugin #4

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Open

mathieu-thomas wants to merge 1 commit into main from codex/explain-how-plugins-work-0fgome

Owner

mathieu-thomas commented Nov 12, 2025

Summary

detect when torchvision lacks CUDA NMS support and hide GPU device options accordingly
add runtime fallback to CPU inference when CUDA NMS errors occur for image, video, or webcam pipelines
update plugin UI defaults and version metadata to reflect safer device handling

Testing

python -m compileall plugins/wan2gp-yolo11


          Handle CUDA NMS fallback in YOLO11 plugin

792fae3

mathieu-thomas added the codex label

— with

ChatGPT Codex Connector

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels