-
Notifications
You must be signed in to change notification settings - Fork 468
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
Onnxrt execution provider configs #992
Conversation
0125e6a
to
9c3a8a7
Compare
Signed-off-by: [email protected] <[email protected]>
Signed-off-by: [email protected] <[email protected]>
9c3a8a7
to
e69d495
Compare
Signed-off-by: [email protected] <[email protected]>
Hi, session.cc files also has several other models (vad,etc), since these are minimal models, they are left as they are for pybind interface, still adding in changes |
Signed-off-by: [email protected] <[email protected]>
Signed-off-by: [email protected] <[email protected]>
Signed-off-by: [email protected] <[email protected]>
Signed-off-by: [email protected] <[email protected]>
Thanks! That sounds good to me. |
Please have a look at errors reported by and the style check by https://github.com/k2-fsa/sherpa-onnx/actions/runs/9665507136 (Note: You can run these two tools locally.) |
Co-authored-by: Fangjun Kuang <[email protected]>
Signed-off-by: [email protected] <[email protected]>
I wanted to keep provider config unique for each models, this would give good control for CUDA and TRT configs, Most of the fixes I had also went away with those patches thats why several bugs |
yes, have been using for cpplint, had a hard time with exit 🙂 |
By the way, you can use
|
Signed-off-by: [email protected] <[email protected]>
Signed-off-by: [email protected] <[email protected]>
Signed-off-by: [email protected] <[email protected]>
Signed-off-by: [email protected] <[email protected]>
I ran the python tests in debug mode and got this error
it would be great to have some hints for fixing this |
Signed-off-by: [email protected] <[email protected]>
Hi @csukuangfj, would be great to have some leads on python fix |
please also change I give you an example below. You need to change other Change sherpa-onnx/sherpa-onnx/python/sherpa_onnx/online_recognizer.py Lines 174 to 183 in 125bb9f
to provider_config = ProviderConfig(provider=provider)
model_config = OnlineModelConfig(
transducer=transducer_config,
tokens=tokens,
num_threads=num_threads,
provider_config=provider_config,
model_type=model_type,
modeling_unit=modeling_unit,
bpe_vocab=bpe_vocab,
debug=debug,
) Remember to change
ProviderConfig,
OnlineModelConfig, |
Signed-off-by: [email protected] <[email protected]>
trt workspace update Co-authored-by: Fangjun Kuang <[email protected]>
Signed-off-by: [email protected] <[email protected]>
Signed-off-by: [email protected] <[email protected]>
Signed-off-by: [email protected] <[email protected]>
Hi @csukuangfj, |
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.
Thank you!
Left some minor comments. Otherwise, it looks good to me.
Signed-off-by: [email protected] <[email protected]>
Hi @csukuangfj, I think it is all good now, expected tests are also passing through, |
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.
Thank you for your contribution!
uffff, Thank you so much |
Config files handling for onnxrt provider
cc : #40