import torch
from transformers import WhisperForConditionalGeneration
model = WhisperForConditionalGeneration.from_pretrained("whisper_finetuned", torch_dtype=torch.float16)
model.save_pretrained("whisper_finetuned")