Skip to content

Commit

Permalink
initial commit to bump v1
Browse files Browse the repository at this point in the history
  • Loading branch information
TParcollet committed Feb 27, 2024
1 parent b0fd3dd commit 455a8d2
Show file tree
Hide file tree
Showing 6 changed files with 6 additions and 6 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
import torch
from app.common import ModelType, get_type
from app.pipelines import Pipeline
from speechbrain.pretrained import EncoderClassifier
from speechbrain.inference import EncoderClassifier


class AudioClassificationPipeline(Pipeline):
Expand Down
2 changes: 1 addition & 1 deletion docker_images/speechbrain/app/pipelines/audio_to_audio.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
import torch
from app.common import ModelType, get_type
from app.pipelines import Pipeline
from speechbrain.pretrained import (
from speechbrain.inference import (
SepformerSeparation,
SpectralMaskEnhancement,
WaveformEnhancement,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
import torch
from app.common import ModelType, get_type
from app.pipelines import Pipeline
from speechbrain.pretrained import EncoderASR, EncoderDecoderASR, WhisperASR
from speechbrain.inference import EncoderASR, EncoderDecoderASR, WhisperASR


class AutomaticSpeechRecognitionPipeline(Pipeline):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@

from app.common import ModelType, get_type
from app.pipelines import Pipeline
from speechbrain.pretrained import GraphemeToPhoneme
from speechbrain.inference import GraphemeToPhoneme


POSTPROCESSING = {ModelType.GRAPHEMETOPHONEME: lambda output: "-".join(output)}
Expand Down
2 changes: 1 addition & 1 deletion docker_images/speechbrain/app/pipelines/text_to_speech.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
import numpy as np
from app.common import ModelType, get_type, get_vocoder_model_id
from app.pipelines import Pipeline
from speechbrain.pretrained import HIFIGAN, FastSpeech2, Tacotron2
from speechbrain.inference import HIFIGAN, FastSpeech2, Tacotron2


class TextToSpeechPipeline(Pipeline):
Expand Down
2 changes: 1 addition & 1 deletion docker_images/speechbrain/requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -3,5 +3,5 @@ starlette==0.27.0
api-inference-community==0.0.32
huggingface_hub>=0.7
transformers==4.30.0
git+https://github.com/speechbrain/speechbrain@v0.5.15
git+https://github.com/speechbrain/speechbrain@v1.0.0
#Dummy.

0 comments on commit 455a8d2

Please sign in to comment.