diff options
| -rw-r--r-- | poetry.lock | 2 | ||||
| -rw-r--r-- | pyproject.toml | 1 | ||||
| -rw-r--r-- | swr2_asr/train.py | 2 |
3 files changed, 3 insertions, 2 deletions
diff --git a/poetry.lock b/poetry.lock index 287e19a..d5a40f1 100644 --- a/poetry.lock +++ b/poetry.lock @@ -999,4 +999,4 @@ files = [ [metadata] lock-version = "2.0" python-versions = "~3.10" -content-hash = "98a9b21411812f0514cbac138756a635b4f9283835791153f467c7f3fe1f4fd1" +content-hash = "db7fb92ac025d1ec61cee65ac267365851f2974063c79c6ba7dcca964c1ba2e8" diff --git a/pyproject.toml b/pyproject.toml index 8490aa5..fdd89a5 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -15,6 +15,7 @@ audioloader = {git = "https://github.com/marvinborner/AudioLoader.git"} tqdm = "^4.66.1" numpy = "^1.25.2" mido = "^1.3.0" +click = "^8.1.7" [tool.poetry.group.dev.dependencies] black = "^23.7.0" diff --git a/swr2_asr/train.py b/swr2_asr/train.py index 29f9372..1ef42aa 100644 --- a/swr2_asr/train.py +++ b/swr2_asr/train.py @@ -1,5 +1,5 @@ """Training script for the ASR model.""" -from AudioLoader.speech.mls import MultilingualLibriSpeech +from AudioLoader.speech import MultilingualLibriSpeech import click import torch import torch.nn as nn |