tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Change the utils.download_emo_models (#199)

Browse Source 
* Change the utils.download_emo_models

Change utils.download_emo_models(config.mirror, model_name, REPO_ID) to utils.download_emo_models(config.mirror, REPO_ID, model_name)

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
Thymustw
2023-11-27 16:49:00 +08:00
committed by GitHub
parent 2c528ce130
commit badb12585a
1 changed files with 162 additions and 161 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
emo_gen.py
View File

@@ -6,6 +6,7 @@ import librosa
import numpy as np import numpy as np
import torch import torch
import torch.nn as nn import torch.nn as nn
from torch.utils.data import Dataset
from torch.utils.data import DataLoader, Dataset from torch.utils.data import DataLoader, Dataset
from tqdm import tqdm from tqdm import tqdm
from transformers import Wav2Vec2Processor from transformers import Wav2Vec2Processor
@@ -108,7 +109,7 @@ def get_emo(path):
wav, sr = librosa.load(path, 16000) wav, sr = librosa.load(path, 16000)
device = config.bert_gen_config.device device = config.bert_gen_config.device
return process_func( return process_func(
np.expand_dims(wav, 0).astype(np.float64), np.expand_dims(wav, 0).astype(np.float),
sr, sr,
model, model,
processor, processor,
@@ -134,7 +135,7 @@ if __name__ == "__main__":
model_name = "./emotional/wav2vec2-large-robust-12-ft-emotion-msp-dim" model_name = "./emotional/wav2vec2-large-robust-12-ft-emotion-msp-dim"
REPO_ID = "audeering/wav2vec2-large-robust-12-ft-emotion-msp-dim" REPO_ID = "audeering/wav2vec2-large-robust-12-ft-emotion-msp-dim"
if not Path(model_name).joinpath("pytorch_model.bin").exists(): if not Path(model_name).joinpath("pytorch_model.bin").exists():
utils.download_emo_models(config.mirror, model_name, REPO_ID) utils.download_emo_models(config.mirror, REPO_ID, model_name)
processor = Wav2Vec2Processor.from_pretrained(model_name) processor = Wav2Vec2Processor.from_pretrained(model_name)
model = EmotionModel.from_pretrained(model_name).to(device) model = EmotionModel.from_pretrained(model_name).to(device)