06c46fb 201ed31 06c46fb
1
2
3
4
5
6
7
import torch from transformers import Wav2Vec2Model, Wav2Vec2Processor from config import MODEL_NAME, DEVICE processor = Wav2Vec2Processor.from_pretrained(MODEL_NAME) feature_extractor = Wav2Vec2Model.from_pretrained(MODEL_NAME).to(DEVICE)
