Skip to content

Conversation

tchawada
Copy link

Added support for "facebook/wav2vec2-base-960h" model via AutoModelForCTC class.

]


def load_CTC_model(model_config):
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

nit: let's keep the CTC as ctc across all test file changes.

Returns:
torch.Tensor: A list of output features generated by the model for each prompt.
"""
seq_len = 480000
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

nit: Keep this in Constants file and fetch it from there

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants