Skip to content
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

Allow users to specify the stoks model? #141

Open
BBC-Esq opened this issue May 20, 2024 · 0 comments
Open

Allow users to specify the stoks model? #141

BBC-Esq opened this issue May 20, 2024 · 0 comments

Comments

@BBC-Esq
Copy link
Contributor

BBC-Esq commented May 20, 2024

Can we allow a user to specify the stoks model? Here's the constructor that allows specifying the s2a and t2s models:

ef __init__(self, t2s_ref=None, s2a_ref=None, optimize=True, torch_compile=False, device=None):

We'd change it to the following but would have to modify other scripts as well presumably:

def __init__(self, t2s_ref=None, s2a_ref=None, stoks_ref=None, optimize=True, torch_compile=False, device=None):
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Development

No branches or pull requests

1 participant