| from transformers import WhisperForConditionalGeneration | |
| import torch | |
| model = WhisperForConditionalGeneration.from_pretrained(".") | |
| processor = WhisperProcessor.from_pretrained("C:/Project/Private/AudioRecord/Spiritbox/Pro/whisper-thai-finetuned/checkpoint-500") | |
| example_input = torch.randn(1, 80, 3000) | |
| traced = torch.jit.trace(model, example_input) | |
| traced.save("whisper_thai_traced.pt") | |
| model.save_pretrained("C:/Project/Private/AudioRecord/Spiritbox/Pro/whisper-thai-finetuned/export") | |
| processor.save_pretrained("C:/Project/Private/AudioRecord/Spiritbox/Pro/whisper-thai-finetuned/export") |