[Hackathon 7th] t2s inference compatible with PIR api #3923
Conversation
This file contains hidden or 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
|
Thanks for your contribution! |
zxcd
reviewed
Dec 3, 2024
| @@ -120,6 +121,15 @@ def parse_args(): | |||
| return args | |||
|
|
|||
|
|
|||
| def get_model_suffix(inference_dir, model_name): | |||
| if os.path.exists(os.path.join(inference_dir, model_name + ".pdmodel")): | |||
There was a problem hiding this comment.
inference support both format support, like:
paddle.inference.Config(model_name_or_path, model_prefix)
model_name_or_path: the root path of model
model_prefix: name of model, without suffix
In this time, interface should be fixed:
[def get_predictor(](
also for support it compatibility, you can switch it with paddle version.
This was referenced Dec 3, 2024
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.
PR types
Others
PR changes
Others
Describe
This pull request includes changes to the
paddlespeech/t2s/exps/inference.pyfile to enhance the flexibility of model file handling. The most important changes involve importing theosmodule, adding a new function to determine the model file suffix, and updating themainfunction to use the new logic for model file suffixes.Enhancements to model file handling:
paddlespeech/t2s/exps/inference.py: Imported theosmodule to enable file existence checks.paddlespeech/t2s/exps/inference.py: Addedget_model_suffixfunction to determine the appropriate suffix for model files based on their existence in the specified directory.paddlespeech/t2s/exps/inference.py: Updated themainfunction to use theget_model_suffixfunction for determining the suffix of theamandvocmodel files dynamically. [1] [2]