Skip to content

Commit

Permalink
Update MozillaDeepSpeechWav2Txt recipe (#202)
Browse files Browse the repository at this point in the history
* Upgrade deepspeech-gpu to 0.9.3

* Upgrade deepspeech-gpu to 0.10.0a3

* Add a comment stating deepspeech-gpu Python version compatibility
  • Loading branch information
this authored Jun 21, 2024
1 parent dbe21a6 commit e1647fd
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions data/mozilla_deepspeech_wav2txt.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,10 +36,10 @@
except ImportError:
from pipes import quote

_global_modules_needed_by_name = ["deepspeech-gpu==0.6.1"]
_global_modules_needed_by_name = ["deepspeech-gpu==0.10.0a3"]

#####Please fill up before usage
MODEL_PATH = "/home/dmitry/Desktop/DAI/deepspeech/deepspeech-0.6.1-models"
MODEL_PATH = "/home/dmitry/Desktop/DAI/deepspeech/deepspeech-0.10.0a3-models"
beam_width = 500 # Beam width for the CTC decoder
lm_alpha = 0.75 # Language model weight (lm_alpha)
lm_beta = 1.85 # Word insertion bonus (lm_beta)
Expand Down Expand Up @@ -67,7 +67,8 @@ def convert_samplerate(audio_path, desired_sample_rate):


class MozillaDeepSpeechWav2Txt(BaseData):
_modules_needed_by_name = ["deepspeech-gpu==0.6.1"]
# Note: deepspeech-gpu only supports upto Python 3.9
_modules_needed_by_name = ["deepspeech-gpu==0.10.0a3"]

@staticmethod
def create_data(X: dt.Frame = None) -> dt.Frame:
Expand Down

0 comments on commit e1647fd

Please sign in to comment.