speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

install requirements automatically

Browse Source
This commit is contained in:
游雁 2024-03-25 11:15:24 +08:00
parent 1df10311c4
commit 5942057698
2 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
funasr/download/download_from_hub.py
View File

@ -73,6 +73,8 @@ def download_from_ms(**kwargs):
if isinstance(kwargs, DictConfig): if isinstance(kwargs, DictConfig):
kwargs = OmegaConf.to_container(kwargs, resolve=True) kwargs = OmegaConf.to_container(kwargs, resolve=True)
if os.path.exists(os.path.join(model_or_path, "requirements.txt")): if os.path.exists(os.path.join(model_or_path, "requirements.txt")):
requirements = os.path.join(model_or_path, "requirements.txt")
print(f"Detect model requirements, begin to install it: {requirements}")
from funasr.utils.install_model_requirements import install_requirements from funasr.utils.install_model_requirements import install_requirements
install_requirements(os.path.join(model_or_path, "requirements.txt")) install_requirements(os.path.join(model_or_path, "requirements.txt"))
return kwargs return kwargs

7
funasr/frontends/whisper_frontend.py
View File

@ -1,8 +1,8 @@
from typing import Tuple from typing import Tuple
import torch import torch
import torch.nn as nn import torch.nn as nn
import whisper
from whisper.audio import HOP_LENGTH, N_FFT, N_SAMPLES
from funasr.register import tables from funasr.register import tables
from torch.nn.utils.rnn import pad_sequence from torch.nn.utils.rnn import pad_sequence
@ -26,7 +26,8 @@ class WhisperFrontend(nn.Module):
super().__init__() super().__init__()
assert fs == 16000 assert fs == 16000
self.fs = fs self.fs = fs
import whisper
from whisper.audio import HOP_LENGTH, N_FFT, N_SAMPLES
self.n_fft = N_FFT self.n_fft = N_FFT
self.win_length = N_FFT self.win_length = N_FFT
self.hop_length = HOP_LENGTH self.hop_length = HOP_LENGTH