Skip to content

Commit

Permalink
Adjusted n_mels change for v3 models only
Browse files Browse the repository at this point in the history
  • Loading branch information
Aleks committed Mar 30, 2024
1 parent 9e75c90 commit c73ace7
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/wordcab_transcribe/services/transcribe_service.py
Original file line number Diff line number Diff line change
Expand Up @@ -89,7 +89,7 @@ def __init__(
)
elif self.model_engine == "tensorrt-llm":
logger.info("Using tensorrt-llm model engine.")
if "distil" in self.model_path:
if "v3" in self.model_path:
n_mels = 128
else:
n_mels = 80
Expand Down

0 comments on commit c73ace7

Please sign in to comment.