a-One-Fan/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Pull Requests Activity

Update wav2vec2 wrapper

Browse Source
This commit is contained in:
James Betker 2022-03-15 11:35:38 -06:00
parent 1e3a8554a1
commit b754058018
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
codes/models/asr/w2v_wrapper.py → codes/models/audio/asr/w2v_wrapper.py
View File

@ -2,12 +2,11 @@ from itertools import groupby
import torch import torch
import torch.nn as nn import torch.nn as nn
import torch.nn.functional as F from transformers import Wav2Vec2ForCTC
from transformers import Wav2Vec2ForCTC, Wav2Vec2CTCTokenizer
from transformers.models.wav2vec2.modeling_wav2vec2 import Wav2Vec2Attention, Wav2Vec2Model from transformers.models.wav2vec2.modeling_wav2vec2 import Wav2Vec2Attention, Wav2Vec2Model
from data.audio.unsupervised_audio_dataset import load_audio from data.audio.unsupervised_audio_dataset import load_audio
from models.tacotron2.text import symbols, sequence_to_text from models.audio.tts.tacotron2.text import sequence_to_text
from trainer.networks import register_model from trainer.networks import register_model
from utils.util import opt_get from utils.util import opt_get