Add support for Mac mps auto detect in demo.py and fix FutureWarning #76
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
imClumsyPanda
commented
Dec 27, 2024
imClumsyPanda
commented
- Add mps auto detect in demo.py when running on Mac
- fix weights_only FutureWarning when calling torch.load
|
@imClumsyPanda Hello! Could you please upload some instructions and screen shots on running on MAC MPs? |
|
Running on Mac with M series chip and check with the gpu usage (the peak shows when mps is used) Inference speed is 2x faster, and 1.5x faster in total. 
|
|
@imClumsyPanda Thanks for contribution! I will test this as soon as possible in next week |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.