Update benchmark.py

benchmark.py

@@ -4,16 +4,19 @@ from transformers import pipeline
 from datasets import load_dataset
 
 # config
-model_id = "kotoba-tech/kotoba-whisper-v1.0"
 generate_kwargs = {"language": "japanese", "task": "transcribe"}
+model_id = "kotoba-tech/kotoba-whisper-v1.0"
+torch_dtype = torch.float32
+device = "cpu"
 
 # load model
 pipe = pipeline(
     "automatic-speech-recognition",
     model=model_id,
-    chunk_length_s=15,
-    batch_size=64
+    torch_dtype=torch_dtype,
+    device=device,
 )
+
 test_audio = [
     "kotoba-whisper-eval/audio/long_interview_1.wav",
     "kotoba-whisper-eval/audio/manzai1.wav",