Skip to content

Enable Visualized_BGE Inference on CPU#937

Merged
JUNJIE99 merged 1 commit intoFlagOpen:masterfrom
hotchpotch:fix_device_visualized
Jul 3, 2024
Merged

Enable Visualized_BGE Inference on CPU#937
JUNJIE99 merged 1 commit intoFlagOpen:masterfrom
hotchpotch:fix_device_visualized

Conversation

@hotchpotch
Copy link
Copy Markdown
Contributor

Currently, it seems to work only in a CUDA environment. This change ensures it will work correctly by specifying the use of the CPU when CUDA is not available.

AttributeError: 'Visualized_BGE' object has no attribute 'device'

This change ensures it will work correctly by specifying the use of the CPU when CUDA is not available.

@JUNJIE99
Copy link
Copy Markdown
Collaborator

JUNJIE99 commented Jul 3, 2024

Thank you for your contribution!

@JUNJIE99 JUNJIE99 merged commit 888f34b into FlagOpen:master Jul 3, 2024
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