Fixed bugs related to sum #693
Merged
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
Signed-off-by: ふぁ <[email protected]>
Signed-off-by: ふぁ <[email protected]>
Signed-off-by: ふぁ <[email protected]>
fa0311
commented
Nov 29, 2024
| if providers: | ||
| for provider in providers or []: | ||
| if provider in available_providers: | ||
| valid_providers.append(provider) |
There was a problem hiding this comment.
Added if providers: like other models.
This was causing unintended fallbacks.
new_session(model_name="sam", providers=["CUDAExecutionProvider","CPUExecutionProvider"])2024-11-29 13:14:59.2810429 [E:onnxruntime:Default, provider_bridge_ort.cc:1848 onnxruntime::TryGetProviderInfo_TensorRT] D:\a\_work\1\s\onnxruntime\core\session\provider_bridge_ort.cc:1539 onnxruntime::ProviderLibrary::Get [ONNXRuntimeError] : 1 : FAIL : LoadLibrary failed with error 126 "" when trying to load "c:\Users\yuki\Documents\School\2024\background-erase\.venv\lib\site-packages\onnxruntime\capi\onnxruntime_providers_tensorrt.dll"
*************** EP Error ***************
EP Error D:\a\_work\1\s\onnxruntime\python\onnxruntime_pybind_state.cc:507 onnxruntime::python::RegisterTensorRTPluginsAsCustomOps Please install TensorRT libraries as mentioned in the GPU requirements page, make sure they're in the PATH or LD_LIBRARY_PATH, and that your GPU is supported.
when using ['CUDAExecutionProvider', 'CPUExecutionProvider', 'TensorrtExecutionProvider', 'CUDAExecutionProvider', 'CPUExecutionProvider']
Falling back to ['CUDAExecutionProvider', 'CPUExecutionProvider'] and retrying.
There was a problem hiding this comment.
This appears to have changed with this commit: 6f1dd31
fa0311
commented
Nov 29, 2024
Comment on lines
+146
to
+155
| prompt = kwargs.get( | ||
| "sam_prompt", | ||
| [ | ||
| { | ||
| "type": "point", | ||
| "label": 1, | ||
| "data": [int(img.width / 2), int(img.height / 2)], | ||
| } | ||
| ], | ||
| ) |
There was a problem hiding this comment.
Added default prompt
This simply selects the center of the image.
The existing one simply causes a validation error, but better than that.
|
Thanks a lot! 👏 |
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.
No description provided.