ASR-Hearing-Impaired-API / convert_fp16.py
Thedeezat's picture
Initial clean API Space
766a353
Raw
History Blame Contribute Delete
217 Bytes
import torch
from transformers import WhisperForConditionalGeneration
model = WhisperForConditionalGeneration.from_pretrained("whisper_finetuned", torch_dtype=torch.float16)
model.save_pretrained("whisper_finetuned")