prepare_ckpt.py 334 Bytes
Newer Older
chenzk's avatar
v1.0  
chenzk committed
1
2
3
4
5
6
7
8
9
10

from transformers import WhisperProcessor, WhisperForConditionalGeneration 
from funasr import AutoModel

device="cuda:0"
model_id = "openai/whisper-large-v3"
processor = WhisperProcessor.from_pretrained(model_id)
model = WhisperForConditionalGeneration.from_pretrained(model_id).to(device)

model = AutoModel(model="paraformer-zh")