-
Notifications
You must be signed in to change notification settings - Fork 118
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.
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
new: allow users to override providers #214
Conversation
available_providers = ort.get_available_providers() | ||
for provider in onnx_providers: | ||
# check providers available | ||
provider_name = provider if isinstance(provider, str) else provider[0] | ||
if provider_name not in available_providers: | ||
raise ValueError( | ||
f"Provider {provider_name} is not available. Available providers: {available_providers}" | ||
) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks for adding the available_providers
check!
I'm pushing a few changes to README with suggestions for user facing API. Would love to hear what you think @joein! |
Hi @NirantK This is an awesome addition but I'm not sure if it is being realized. Have attached the collab link here - https://colab.research.google.com/drive/1I4Z547eE0utHmp2pxIgaLx0PXQhNFWdj?usp=sharing Thanks a lot! |
Hey @GautamR-Samagra, for some reason CUDAExecutionProvider has not been set in your notebook, we'll investigate it, thanks for sharing! |
Supersedes #214, closes #52 and #97